Leetcode 219

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

方法:

一开始想到的最自然的方法是采用两个循环,但是直觉告诉我肯定会超过时间复杂度的上限。因此考虑选择使用其他的方法,想到采用之前使用的字典的方法,将列表的数值作为键值,索引值作为实际的值。然后对列表进行遍历,遇到相同的就计算索引差是否小于k。如果超过就更新键值对,继续遍历操作

最终的AC代码如下:

class Solution:
    def containsNearbyDuplicate(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: bool
        """
        dic = {}
        for i in range(len(nums)):
            if nums[i] not in dic:
                dic[nums[i]] = i
            else:
                if abs(dic[nums[i]] - i) <= k:
                    return True
                else:
                    dic[nums[i]] = i
        return False

猜你喜欢

转载自blog.csdn.net/jhlovetll/article/details/85682845