思想:
1.先判断nums是否为空,为空则返回false。再判断k是否等于0,为0则返回false。
2.由于range函数后面的范围取不到则需要k+1,然后需要判断加1后的k有什么超过nums的长度。若超过,则选择nums的长度作为k值。
3.定义了一个双向队列windows来存储元素的下标,一个集合d来存储元素。
4.先从下标0开始取元素并判断是否已经存在d中,由于我们的下标范围是0-k+1,若存在d中,则说明有满足条件的i和j。若不存在则将该元素加入d,依次循环直到结束。
5.若此次循环并没有找到,则开始第二个循环。由于要满足i和j的绝对值最大为k,每次循环前我们需要删除d中一个元素,我们可以根据windows所存下标依次删。然后在判断之后的元素是否在d中,若存在d中,则说明有满足条件的i和j。若不存在则将该元素加入d,依次循环直到结束。
from collections import deque
class Solution:
def containsNearbyDuplicate(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: bool
"""
if not nums:
return False
if k == 0:
return False
k = k + 1
k = min(k, len(nums))
window = deque([])
d = set()
for i in range(0, k):
if nums[i] in d:
return True
d |= {nums[i]}
window.append(i)
for i in range(k, len(nums)):
d -= {nums[window.popleft()]}
if nums[i] in d:
return True
d |= {nums[i]}
window.append(i)
return False
大佬的思想,学习学习