leetcode 219. 存在重复元素 II

这道题我首先是用的暴力法，超时。

下面这个方法是用个map保存 数和它的下标。这个方法通过了，但是用时需要106ms

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        map<int,int> mapIndex;//<数值,下标>
        for(int i = 0 ;i<nums.size() ;i++)
        {
            //在map里找到当前的数，并且当前的数的下标和前一个相同的数的下标之差小于等于k
            if(mapIndex.find(nums[i])!=mapIndex.end() && (i-mapIndex[nums[i]])<=k )
                return true;
            else
            //否则更新map里当前这个数对应的下标值
                mapIndex[nums[i]] = i;
        }
        return false;
    }
};

下面这个方法我是看评论区学着使用set，这个set维护k个数，如果能在set里找到当前的数，则true，否则把当前这个数加入set，判断set的数据个数有没有超过k，超出的话就移除最开头加入set的数。

我这样写用时116ms，虽然通过了，但还是很慢的方法。

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