要求
给定一个整数数组和一个整数 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
解题思路:这个题是217题的延续,我们使用dict对数据的index进行存储,当遇到相同元素的时候,我们比较两个元素index之差是不是不大于k,不大于返回True,循环完成,没有符合条件的,返回False.