题目
July 10, 2021 · View on GitHub
给定一个整数数组和一个整数 k,判断数组中是否存在两个不同的索引 i 和 j,使得 nums [i] = nums [j],并且 i 和 j 的差的 绝对值 至多为 k。
示例 1:
输入: nums = [1,2,3,1], k = 3
输出: true
示例 2:
输入: nums = [1,0,1,1], k = 1
输出: true
示例 3:
输入: nums = [1,2,3,1,2,3], k = 2
输出: false
参考答案
class Solution {
public:
bool containsNearbyDuplicate(vector<int>& nums, int k) {
unordered_set<int> existed;
int n = nums.size();
int curr = 0;
for (int i = 0; i < n; ++i)
{
curr = nums[i];
if (existed.find(curr) == existed.end())
{
existed.insert(curr);
if (existed.size() > k)
{
existed.erase(nums[i-k]);
}
}
else
{
return true;
}
}
return false;
}
};