问题描述:
给定一个整数数组和一个整数 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
基本思路:
运用滑动窗口+哈希集的思想。
一开始在滑动窗口没有建立完的情况下,特殊判断。
在已经建立完毕的情况下,不断删除和插入元素。看看是否有重复的。
AC代码:
class Solution {
public:
bool containsNearbyDuplicate(vector<int>& nums, int k) {
int i = 0, j = 0;
set<int> s;
while (j != nums.size()) {
if (s.size() != k + 1) { // 处理刚开始的情况
if (s.count(nums[j])) { // 刚开始就有重复元素的情况
return true;
} else {
s.insert(nums[j++]);
}
} else { // 滑动窗口已经建立的情况下
s.erase(nums[i++]);
if (s.count(nums[j])) {
return true;
} else {
s.insert(nums[j++]);
}
}
}
return false;
}
};