【LeetCode】219.存在重复元素II

219.存在重复元素II

知识点：数组；双指针；

题目描述

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

示例

输入：nums = [1,2,3,1], k = 3
输出：true

输入：nums = [1,0,1,1], k = 1
输出：true

输入：nums = [1,2,3,1,2,3], k = 2
输出：false

---

解法一：哈希表

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

解法二：滑动窗口

因为要满足两个下标在k范围内，索性可以维持一个长度为k的滑动窗口，然后检查这个窗口内是否有重复的元素。所以直接用维持一个长度为k的集合set，然后不断把元素加到集合里，不断把元素移出集合；

class Solution:
    def containsNearbyDuplicate(self, nums: List[int], k: int) -> bool:
        s = set()
        for i, num in enumerate(nums):
            if num in s:
                return True
            s.add(num)
            if len(s) > k:
                s.remove(nums[i-k])
        return False