【LeetCode】第219题——存在重复元素II（难度：简单）

题目描述

给定一个整数数组和一个整数 k，判断数组中是否存在两个不同的索引 i 和 j，使得 nums [i] = nums [j]，并且 i 和 j 的差的 绝对值 至多为 k。

1. 示例 1:
   输入: nums = [1,2,3,1], k = 3
   输出: true

2. 示例 2:
   输入: nums = [1,0,1,1], k = 1
   输出: true

3. 示例 3:
   输入: nums = [1,2,3,1,2,3], k = 2
   输出: false

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/contains-duplicate-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

思路一：HashMap，key为数值，value为最近一次出现的位置，当遍历的位置 i 与对应的value之差小于等于给定值 k ，则返回true。

思路二：散列表，最多也只会开辟容量为 k 的空间。利用Set集合，如果遍历的数字不存在Set里，就add进去，若此时size超过了 k ，则把前第 k 个添加进来的值remove掉，如果遍历的数字在Set中，直接返回true即可，因为下标之差绝对不超过 k 。

代码详解

思路一：HashMap

class Solution {
    public boolean containsNearbyDuplicate(int[] nums, int k) {
        if(nums.length < 2) {	// 如果都没有两个元素，也就谈不上相等
            return false;
        }
        HashMap<Integer, Integer> hm = new HashMap<>();
        for(int i = 0; i < nums.length; ++i) {
            if(hm.containsKey(nums[i])) {
                if(i - hm.get(nums[i]) <= k) {	// 如果存在，则看看下标差是否超过k
                    return true;
                } else {	// 若超过则更新value，即更新至最近一次出现的位置
                    hm.replace(nums[i], i);
                }
            } else {
                hm.put(nums[i], i);	// 如果不存在就直接添加进HashMap里，value是下标
            }
        }
        return false;
    }
}

思路二：散列表（推荐）

class Solution {
    public boolean containsNearbyDuplicate(int[] nums, int k) {
        if(nums.length < 2) {	// 如果都没有两个元素，也就谈不上相等
            return false;
        }
        Set<Integer> hs = new HashSet<>();
        for(int i = 0; i < nums.length; ++i) {
            if(hs.contains(nums[i])) {	// 如果存在直接返回true，因为下标之差绝对小于k
                    return true;
            }
            hs.add(nums[i]);	// 不存在就添加
            if(hs.size() > k) {	// 这里限制了集合大小不会超过k，也就决定了下标差的最大值
                hs.remove(nums[i - k]);	// 前第k个添加进集合里的元素被移除
            }
        }
        return false;
    }
}

注意点

• 思路二的散列表比思路一的哈希表开辟空间少，算法思路也容易理解，推荐思路二。