给你一个整数数组 nums 和一个整数 k ,判断数组中是否存在两个 不同的索引 i 和 j ,满足 nums[i] == nums[j] 且 abs(i - j) <= k 。如果存在,返回 true ;否则,返回 false 。
解法①:直接法,超出时间限制
class Solution:
def containsNearbyDuplicate(self, nums: List[int], k: int) -> bool:
for i in range(len(nums)):
num_one = nums[i]
num_one_index = i
next_index = num_one_index + 1
temp_nums = nums[next_index:]
for j in range(len(temp_nums)):
num_two = temp_nums[j]
if num_one == num_two and abs(i-next_index-j) <= k:
return True
return False
解法②: 哈希表存储重复元素最大下标
class Solution:
def containsNearbyDuplicate(self, nums: List[int], k: int) -> bool:
#哈希表存储当前元素最大下标
dict = {}
for i,num in enumerate(nums):
if num in dict and i-dict[num]<= k:
return True
dict[num] = i
return False
#复杂度
时间复杂度:O(n),其中 n 是数组 nums 的长度。需要遍历数组一次,对于每个元素,哈希表的操作时间都是 O(1)。
空间复杂度:O(n),其中 n 是数组 nums 的长度。需要使用哈希表记录每个元素的最大下标,哈希表中的元素个数不会超过 n
解法③:滑动窗口 ,详解见leetcode官方
class Solution:
def containsNearbyDuplicate(self, nums: List[int], k: int) -> bool:
#滑动窗口大小为k+1
#滑动窗口结束下标为i,则开始下标为max(0,i-k)
#创建哈希集合存储窗口
s = set()
for i,num in enumerate(nums):
#如果当前下标超出窗口则删除窗口内第一个元素
if i>k:
s.remove(nums[i-k-1])
if num in s:
return True
s.add(num)
return False
#复杂度
时间复杂度:O(n),其中 n 是数组 nums 的长度。需要遍历数组一次,对于每个元素,哈希集合的操作时间都是 O(1)。
空间复杂度:O(k),其中 k 是判断重复元素时允许的下标差的绝对值的最大值。需要使用哈希集合存储滑动窗口中的元素,任意时刻滑动窗口中的元素个数最多为 k+1 个