题解 | #奶牛排队问题#

知识点

哈希

解题思路

使用哈希表来存放上次数字出现的下标，如果当前下标减去上次下标小于等于k，表示相邻两个小于k。

注意在获取上次出现下标时，如果之前没有出现过需要给一个默认值，默认值为int的最小值加上n。

Java题解

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param weights int整型一维数组 
     * @param n int整型 
     * @param k int整型 
     * @return bool布尔型
     */
    public boolean checkDuplicate (int[] weights, int n, int k) {
        // write code here
        Map<Integer,Integer> map = new HashMap<>();
        for(int i = 0; i < n; i++){
            int preIndex = map.getOrDefault(weights[i],Integer.MIN_VALUE + n);
            if(i - preIndex <= k) {
                return true;
            }
            map.put(weights[i],i);
        }
        return false;
    }
}