leetcode219. 存在重复元素 II

52Tiramisu2023-08-15 8:36

题目

给你一个整数数组 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

思路：

滑动窗口。借鉴如下思路（未看她代码）

小姐姐刷题-Leetcode 219 Contain duplicate II_哔哩哔哩_bilibili

设置一个set作为窗口，从前往后遍历，如果该值没有在窗口中，就加入，如果加入之后窗口的大小大于k，就踢出去set最开头的那个元素。

代码：

cpp 复制代码

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        set<int> st;
        int i = 0;
        int len = nums.size();
        for(i = 0; i < len; i++){
            int tmp = nums[i];  // 存储当前元素
            if(st.find(tmp) != st.end()){ // 找到相同的元素了
                return true;
                break;
            } else{
                st.insert(tmp);     // 将该元素加入窗口（set）中
                if(st.size() > k){  // 如果加入之后，窗口大小大于k，即不满足 abs(i-j)<=k
                    st.erase(nums[i-k]);    // 就删掉这个窗口中最开头的元素
                }
            }
        }
        // 如果之前没有返回，就说明数组中没有满足条件的值。就返回false
        return false;
    }
};

知识点

滑动窗口。