Leetcode 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

思路解析

把数组中每个元素，以元素：下标的形式存入map中

如果map中已经包含该元素的key,那么判断下标是否满足条件，满足条件直接返回true

如果不满足条件，那么将当前key对应值替换成当前元素下标

因为i-j>k，此时如果j不动，i继续增大，那么永远都不会满足条件

所以只有j变大才可能满足条件

代码

class Solution {
    public boolean containsNearbyDuplicate(int[] nums, int k) {
        
        Map map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            if (!map.containsKey(nums[i])) {
                map.put(nums[i], i);

            } else {
                int j = map.get(nums[i]);
                if (i - j <= k) {
                    return true;
                } else {
                    map.put(nums[i], i);
                }
            }
        }
        // 如果没有找到重复项，则返回false
        return false;
    }
}

如下是大佬思路

维护一个哈希表，里面始终最多包含 k 个元素，当出现重复值时则说明在 k 距离内存在重复元素
每次遍历一个元素则将其加入哈希表中，如果哈希表的大小大于 k，则移除最前面的数字

class Solution {
    public boolean containsNearbyDuplicate(int[] nums, int k) {
        //维护一个不包含重复元素的set，size最大为k
        Set set = new HashSet<>();
        for (int i = 0; i < nums.length; i++) {
            if (!set.add(nums[i])) {
                // 如果添加失败（即nums[i]已经存在于set中），则返回true
                return true;
            }else{
                if(set.size()>k){
                    //移除set首位元素，由于set没有移除首位元素方法，所以根据值来移除
                    set.remove(nums[i-k]);
                }
            }
        }
        // 如果没有找到重复项，则返回false
        return false;

    }
}