Leetcode : 存在重复元素 II

存在重复元素 II

---

给定一个整数数组和一个整数 k，判断数组中是否存在两个不同的索引 i 和 j，使得 nums [i] = nums [j]，并且 i 和 j 的差的绝对值最大为 k。

示例 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

思路是依次遍历，找到同样的元素时，比较index，如果不满足，则更新为最新的index即可。

Python

class Solution:
    def containsNearbyDuplicate(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: bool
        """
        tmp = dict()
        for i in range(len(nums)):
            if tmp.get(nums[i]) is not None:
                if i - tmp.get(nums[i]) <= k:
                    return True
                else:
                    tmp[nums[i]] = i
            else:
                tmp[nums[i]] = i
        return False