刷题日记-219. 存在重复元素 II

80 阅读1分钟

Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情

一、题目描述

219. 存在重复元素 II - 力扣(LeetCode) (leetcode-cn.com)

给你一个整数数组 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 <= 10^5
  • -109 <= nums[i] <= 10^9
  • 0 <= k <= 105

二、思路分析

通过set来记录在[l...l+k]这个区间内的元素,如果范围超出了,就去掉当前索引-k这一个元素,也就是区间的第一个元素,从而保证了滑动窗口的的大小固定

三、AC 代码

class Solution {
    public boolean containsNearbyDuplicate(int[] nums, int k) {
        int n = nums.length;
        

        Set<Integer> set = new HashSet<>();

        for( int i = 0; i < n; i ++){
            if(set.contains(nums[i]))
                return true;
            set.add(nums[i]);

            if(set.size() == k + 1)
                set.remove(nums[i-k]);

        }

        return false;
    }
}

四、总结

这个问题用 set 可以更快,因为字典查找是 log(n) 级别。

范文参考

画解算法:219. 存在重复元素 II - 存在重复元素 II - 力扣(LeetCode) (leetcode-cn.com)

【i和j的差的绝对值最多为k】->window=【i,i+k】 - 存在重复元素 II - 力扣(LeetCode) (leetcode-cn.com)

存在重复元素 II - 存在重复元素 II - 力扣(LeetCode) (leetcode-cn.com)