【力扣】219. 存在重复元素 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
提示：
1 <= nums.length <= 105
-109 <= nums[i] <= 109
0 <= k <= 105

思路一

暴力解法，使用两层for循环遍历所有元素，判断num[i]==num[j] && Math.abs(i-j)<=k。时间复杂度为On²，能解题目，但是力扣无法通过，会超时。

代码

class Solution {
    public boolean containsNearbyDuplicate(int[] nums, int k) {
        for (int i = 0; i < nums.length - 1; i++) {
            for (int j = i + 1; j < nums.length; j++) {
                if (nums[i] == nums[j] && Math.abs(i - j) <= k) {
                    return true;
                }
            }
        }
        return false;
    }
}

思路二

滑动窗口，使用left，right两个指针维护一个大小为k的窗口，使用set来判断窗口内是否有相同的元素。

代码

class Solution {
    public boolean containsNearbyDuplicate(int[] nums, int k) {
    //用来判断是否有相同元素
        Set<Integer> set = new HashSet<>();
        int left = 0;
        for (int right = 0; right < nums.length; right++) {
        //如果窗口大小大于k了，则左指针前移，缩小窗口大小，
            if (Math.abs(left - right) > k) {
                set.remove(nums[left]);
                left++;
            }
            //如果当前的窗口内有相同的元素，则返回true
            if (set.contains(nums[right])) {
                return true;
            } else {
            //没有相同的元素就添加到set中
                set.add(nums[right]);
            }
        }
        return false;
    }
}