LeetCode 219. Contains Duplicate II

维护一个set，保存i之前且与i相差不超过k的一系列数，所以当i>k时，会删去set中nums[i - k - 1]的值，因此k要进行处理，如果k>nums.size()，k赋值为nums.size() - 1。

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        if(k <= 0) return false;
        if(k >= nums.size()) k = nums.size() - 1;
        
        unordered_set<int> num_set;
        
        for(int i = 0; i < nums.size(); ++i){
            if(i > k) num_set.erase(nums[i - k - 1]);
            if(num_set.find(nums[i]) != num_set.end()) return true;
            num_set.insert(nums[i]);
        }
        return false;
    }
};

 

维护map，对应关系是数值及其位置，存放数值的最新位置（即最大位置），如果读入的数已存在且位置之差不大于k，return true。

会慢一些。

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        map<int, int> map;
        for(int i = 0; i < nums.size(); ++i){
            if(map.count(nums[i]) && (i - map[nums[i]]) <= k) return true;
            map[nums[i]] = i;
        }
        return false;
    }
};