给定一个整数数组和一个整数 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
思路:题意是,存在nums [i] = nums [j],并且 i 和 j 的差的绝对值<= k,就返回True,注意:是存在,不是任意。
class Solution:
def containsNearbyDuplicate(self, nums: List[int], k: int) -> bool:
map = {} #num->index
for i in range(len(nums)):
if nums[i] not in map.keys():
map[nums[i]] = i
else:
if i - map[nums[i]] <= k:
return True
else:
map[nums[i]] = i
return False
网友评论