给定一个整数数组和一个整数 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/contains-duplicate-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:
观察题目可以发现,这是一个典型的滑动窗口类型题目,窗口的大小限制为k, 那么我们在每个窗口内寻找是否有遍历到的数,如果有说明满足条件,返回true,如果没有继续遍历,直到遍历结束,set可以去重,所以很好适应这道题目,当然map也可以,使用vector的话会发生超时,然而并不清楚原因,毕竟时间复杂度还是一样的。代码如下:
class Solution {
public:
bool containsNearbyDuplicate(vector<int>& nums, int k) {
int n = nums.size();
// 如果数组大小小于等于1
if(n <= 1) {
return false;
}
// set 来存储不重复的结果
unordered_set<int> arr;
for(int i = 0; i < n; i ++) {
// 如果在set中找到
if(arr.find(nums[i]) != arr.end()) {
return true;
}
// 插入
arr.insert(nums[i]);
// 如果超过了滑动窗口的大小
if(arr.size() > k) arr.erase(nums[i - k]);
}
return false;
}
};
/*作者:heroding
链接:https://leetcode-cn.com/problems/contains-duplicate-ii/solution/chua-dong-chuang-kou-set-by-heroding-qh93/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。*/