给你一个整数数组 nums
和一个整数 k
,判断数组中是否存在两个 不同的索引 i
和 j
,满足 nums[i] == nums[j]
且 abs(i - j) <= k
。如果存在,返回 true
;否则,返回 false
。
示例 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_map<int ,int> mp; // 创建一个哈希表,用于存储数字及其出现的最新位置
for(int i=0;i<nums.size();i++){ // 遍历数组
if(mp.count(nums[i]) && i-mp[nums[i]]<=k) // 如果当前数字在哈希表中出现过且距离上一次出现的位置不超过k,则返回true
return true;
mp[nums[i]]=i; // 将当前数字插入哈希表中,更新其最新位置
}
return false; // 遍历完数组后仍没有找到符合条件的,则返回false
}
};
解法二:滑动窗口
class Solution {
public:
bool containsNearbyDuplicate(vector<int>& nums, int k) {
unordered_set<int> s; // 创建一个无序集合,用于存储最近k个元素
int length = nums.size();
for (int i = 0; i < length; i++) { // 遍历数组
if (i > k) { // 如果超过了k个元素,则删除最旧的元素
s.erase(nums[i - k - 1]);
}
if (s.count(nums[i])) { // 如果当前元素已经在集合中出现过,则返回true
return true;
}
s.emplace(nums[i]); // 将当前元素插入集合中
}
return false; // 遍历完数组后仍没有找到符合条件的,则返回false
}
};
这个代码使用了一个无序集合来存储最近k个元素,通过不断地插入和删除元素来维护集合的大小。在遍历过程中,如果某个元素已经在集合中出现过,则返回true,表示找到了满足条件的重复元素。如果遍历完数组后仍没有找到符合条件的,则返回false。这种方法的时间复杂度为O(n),空间复杂度为O(k)。