LeetCode 219. 存在重复元素 II

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 <= 10^5
• -10^9 <= nums[i] <= 10^9
• 0 <= k <= 10^5

解法：哈希表

从左到右遍历数组 nums，当遍历到下标 i 时，如果存在下标 j<i 使得 nums[i]=nums[j]，则当 i−j≤k 时即找到了两个符合要求的下标 j 和 i。

如果在下标 i 之前存在多个元素都和 nums[i] 相等，为了判断是否存在满足 nums[i]=nums[j] 且 i−j≤k 的下标 j，应该在这些元素中寻找下标最大的元素，将最大下标记为 j，判断 i−j≤k 是否成立。

如果 i−j≤k，则找到了两个符合要求的下标 j 和 i；如果 i−j>k，则在下标 i 之前不存在任何元素满足与 nums[i] 相等且下标差的绝对值不超过 k。

因此，当遍历到下标 i 时，如果在下标 i 之前存在与 nums[i] 相等的元素，应该在这些元素中寻找最大的下标 j，判断 i−j≤k 是否成立。

可以使用哈希表记录每个元素的最大下标。从左到右遍历数组 nums，当遍历到下标 i 时，进行如下操作：

如果哈希表中已经存在和 nums[i] 相等的元素且该元素在哈希表中记录的下标 j 满足 i−j≤k，返回 true；

将 nums[i] 和下标 i 存入哈希表，此时 i 是 nums[i] 的最大下标。

上述两步操作的顺序不能改变，因为当遍历到下标 i 时，只能在下标 i 之前的元素中寻找与当前元素相等的元素及该元素的最大下标。

当遍历结束时，如果没有遇到两个相等元素的下标差的绝对值不超过 k，返回 false。

Java版：

class Solution {
    public boolean containsNearbyDuplicate(int[] nums, int k) {
        int ans = -1;
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            if (map.containsKey(nums[i]) && i - map.get(nums[i]) <= k) {
                return true;
            }
            map.put(nums[i], i);
        }
        return false;
    }
}

Python3版：

class Solution:
    def containsNearbyDuplicate(self, nums: List[int], k: int) -> bool:
        dict_ = {}
        for i in range(len(nums)):
            if nums[i] in dict_ and i - dict_[nums[i]] <= k:
                return True
            dict_[nums[i]] = i
        return False

复杂度分析

• 时间复杂度：O(n)，其中 n 是数组 nums 的长度。需要遍历数组一次，对于每个元素，哈希表的操作时间都是 O(1)。
• 空间复杂度：O(n)，其中 n 是数组 nums 的长度。需要使用哈希表记录每个元素的最大下标，哈希表中的元素个数不会超过 n。