题目
我的解答
依旧使用查找表法,但和LeetCode 217. 存在重复元素 不同的地方在于,217题只需要判断“是否存在重复元素”,无需考虑“如果有重复元素,但不符合条件,还有没有另一个重复元素?”而这道题就需要考虑这个问题。例如示例2,第一个1和第2个1就不满足条件,但第2个1和第3个1就满足了。
思路:设置一个字典,key存放第i个元素,value存放元素在数组里的位置。如果遇到不满足条件的重复元素,那么当下次再遇到时,就把原先那个元素对应字典里的value改为新的重复元素在数组里的位置。
nums = [1,0,1,1]
k = 3
n = len(nums)
nums_0 = nums[0]
dic = {}
dic[nums_0] = 0
loc = -1
for i in range(1,n):
if nums[i] in dic.keys():
loc = dic.get(nums[i])
if i-loc <= k:
return True
else:
dic[nums[i]] = i
continue
dic[nums[i]] = i
return False
比较好的答案
用的就是查找表法。