219. 存在重复元素II

问题链接

https://leetcode.cn/problems/contains-duplicate-ii/description/

解题思路

这道题目是一个经典的滑动窗口题。

常规解法，注意边界值就行。

注意我们应该完全模拟滑动窗口，以防出现复杂度O(n的平方)的情况。

代码

class Solution:
    def containsNearbyDuplicate(self, nums: List[int], k: int) -> bool:
        if len(nums) < 2:
            return False
        cur_set = set()
        for i in range(len(nums)):
            if i > k:
                cur_set.remove(nums[i-k-1])
            if nums[i] not in cur_set:
                cur_set.add(nums[i])
            else:
                return True
        return False

 代码二：

class Solution:
    def containsNearbyDuplicate(self, nums, k: int) -> bool:
        if len(nums) < 2:
            return False
        if k >= len(nums):
            return len(set(nums)) != len(nums)
        mem_set = set(nums[:k])
        if len(mem_set) < k:
            return True
        for i in range(k, len(nums)):
            if nums[i] in mem_set:
                return True
            mem_set.add(nums[i])
            mem_set.remove(nums[i - k])
        return False