leetcode_219 存在重复元素 II

47 阅读1分钟

要求

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

核心代码

class Solution:
    def containsNearbyDuplicate(self, nums: List[int], k: int) -> bool:
        record = dict()
        for i,num in enumerate(nums):
            if record.get(num,-1) != -1:
                if i - record[num] <= k:
                    return True
            record[num] = i
        return False

image.png

解题思路:这个题是217题的延续,我们使用dict对数据的index进行存储,当遇到相同元素的时候,我们比较两个元素index之差是不是不大于k,不大于返回True,循环完成,没有符合条件的,返回False.