LeetCode 219. 存在重复元素 II(python)

题目链接

题目描述:

给定一个整数数组和一个整数 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表示:索引的差值小于等于k
将数组的索引和值作为value和key放入字典中,
如果当前值出现过,判断当前值的索引与字典中的索引差值是否小于等于k,若是,返回True
如果不是,或者没有出现过,将当前值与其索引放入字典中。

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

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/weixin_44740082/article/details/92174439