leetcode219. 存在重复元素 II

题目

给你一个整数数组 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最开头的那个元素。

代码:

class Solution {
public:
    bool containsNearbyDuplicate(vector& nums, int k) {
        set 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;
    }
};

 知识点

滑动窗口。

你可能感兴趣的:(LeetCode,算法,数据结构)