题解 | #奶牛排队问题# 哈希表

知识点

哈希表

思路

开一个哈希表记录每个体重的牛上一次出现的index, 每次出现相同体重的牛就看一下是否符合距离<=k

符合则返回true

更新当前体重的牛的位置

时间复杂度为

AC Code (C++)

#include <unordered_map>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param weights int整型vector 
     * @param n int整型 
     * @param k int整型 
     * @return bool布尔型
     */
    bool checkDuplicate(vector<int>& weights, int n, int k) {
        // 哈希表记录最后出现的位置
        unordered_map<int, int> mp;
        for (int i = 0; i < n; i ++) {
            if (mp.count(weights[i])) {
                if (i - mp[weights[i]] <= k) return true;
            }
            mp[weights[i]] = i;
        }
        return false;
    }
};