- 题目描述
给定一个整数数组和一个整数 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的要求,所以测试用例[99,99] k=2,输出为false,报错
- 看评论才知道这个问题,然后修改了源代码,但是因为还是用的遍历的操作,显示超时。
- 网上大佬们的做法:依次遍历数组中的每个元素,如果是没有出现过的,就把当前元素的值和索引存在一个字典中,如果是出现过的元素,那么就判断当前索引与字典中的那个也是这个值的元素,其索引与当前索引的差值是否小于等于k,如果是的话,就返回True。
- 代码一 超时
class Solution:
def containsNearbyDuplicate(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: bool
"""
if len(nums) == 0:
return False
for i in range(len(nums)-1):
for j in range(i+1,len(nums)):
if nums[i] == nums[j]:
if (j-i <= k) or (i - j >= -k):
return True
return False
- 代码二
class Solution:
def containsNearbyDuplicate(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: bool
"""
numsDict = {}
for i in range(len(nums)):
if nums[i] in numsDict and i - numsDict[nums[i]] <= k:
return True
else:
numsDict[nums[i]] = i
return False