219 Contains Duplicate II

Contains Duplicate II
判断数组中是否存在距离小于k的两个相同元素

讨论区中一个c++答案:

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k)
    {
       unordered_set<int> s;

       if (k <= 0) return false;
       if (k >= nums.size()) k = nums.size() - 1;

       for (int i = 0; i < nums.size(); i++)
       {
           if (i > k) s.erase(nums[i - k - 1]);//保持set中仅有k个元素
           if (s.find(nums[i]) != s.end()) return true;
           s.insert(nums[i]);
       }

       return false;
    }
};

使用map:

bool containsNearbyDuplicate(vector<int>& nums, int k) {

    unordered_map<int,int> nmap;
    for (int i = 0; i <nums.size();i++)
    {
        if (nmap.count(nums[i]) == 0)
            nmap[nums[i]] = i;
        else if (i - nmap[nums[i]] <=k)
            return true;
        else
            nmap[nums[i]] = i;//重复元素距离大于k时更新该元素为后者
    }

    return false;
阅读更多
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页