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;