哈希表 -leetcode-219. 存在重复元素II

219. 存在重复元素 II

题目描述

给你一个整数数组 nums 和一个整数 k ，判断数组中是否存在两个 不同的索引 i 和 j ，满足 nums[i] == nums[j] 且 abs(i - j) <= k 。如果存在，返回 true ；否则，返回 false 。

提示：

• $1<=nums.length<=10^5$
• $10^9<=nums[i]<=10^9$
• $0<=k<=10^5$

暴力（tle）

数据范围有点大 $1<=nums.length<=10^5$，两层for暴力会 tle.

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

哈希表

类似于 拉链法，把 nums中相同元素对应的下标 index 放在同一个 “桶”中。

思路： 🤔

• 用空间换时间，使用哈希表存放$nums$中每个元素出现的下标；
  • key：$nums$中的元素 num；
  • value：$num$ 在nums中的index；
• 遍历map中所有key，查看map的 value对应的list中所有相邻两个值是否 <= k

注意：一开始审题有误，需要查看所有相同元素的下标；而不仅仅是查看 nums中 前两个相同元素的 index是否符合条件。

class Solution {
    public boolean containsNearbyDuplicate(int[] nums, int k) {
        int n = nums.length;
        // key:num; value: num在nums中的index
        Map<Integer, List<Integer>> map = new HashMap<>();
        for (int i = 0; i < n; i++) {
            if (!map.containsKey(nums[i])) {
                List<Integer> list = new ArrayList<>();
                list.add(i);
                map.put(nums[i], list);
            } else {
                List<Integer> list = map.get(nums[i]);
                list.add(i);
                map.put(nums[i], list);
            }
        }
        // 遍历map中所有key，查看map的 value对应的list中所有相邻两个值是否 <= k
        for (int key : map.keySet()) {
            List<Integer> list = map.get(key);
            if (list.size() < 2) continue;
            for (int i = 1; i < list.size(); i++) {
                if (list.get(i) - list.get(i - 1) <= k) 
                    return true;
            }
        }
        System.out.println(map);
        return false;
    }
}

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$