219. 存在重复元素 II

 

思路

难度简单506收藏分享切换为英文接收动态反馈

给你一个整数数组 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

 

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