滑动窗口系列-Leetcode 219. 存在重复元素 II

问题描述

给定一个整数数组和一个整数 k，判断数组中是否存在两个不同的索引 i 和 j，使得 nums [i] = nums [j]，并且 i 和 j 的差的 绝对值 至多为 k。

示例 1:

输入: nums = [1,2,3,1], k = 3
输出: true

解题报告

维持一个大小为k的窗口，将窗口内的数放到集合中，一直检查窗口内是否相同元素，如果存在就返回 true，否则返回 false。

• 移动右指针，检查窗口内的元素放到集合中，检查是否存在重复元素；
• 如果窗口内的元素个数大于 k，移动左指针；
• 移动左指针，将窗口移出的元素从集合中删除；

实现代码

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        int l=0,r=0;
        set<int>s;
        while(r<nums.size()){
            if(r-l>k){
                s.erase(nums[l]);
                l++;
            }
            if(s.find(nums[r])!=s.end())return true;
            s.insert(nums[r++]);
        }
        return false;
    }
};

参考资料

[1] Leetcode 219. 存在重复元素 II