给你一个整数数组 nums 和一个整数 k ,判断数组中是否存在两个 不同的索引 i 和 j ,满足 nums[i] == nums[j] 且 abs(i - j) <= k 。如果存在,返回 true ;否则,返回 false 。
示例 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
哈希表
当遍历到下标 i 时,如果在下标 i 之前存在与 nums[i] 相等的元素(如果使用哈希表来完成这个,时间复杂度为O(1)),应该在这些元素中寻找最大的下标 j ,判断 i - j <= k 是否成立。可以使用哈希表记录每个元素的最大下标。从左到右遍历数组 ,当遍历到下标 i 时,进行如下操作:
如果哈希表中已经存在nums[i] 这个键,且该键记录的下标 j 满足 i - j <= k,返回 True;
否则将 num[i]: i 存入哈希表,此时 i 是 nums[i] 的最大下标。
当遍历结束时,如果没有遇到两个相等元素的下标差的绝对值不超过 k,返回False 。
时间复杂度O(n)
空间复杂度O(n)
class Solution:
def containsNearbyDuplicate(self, nums: List[int], k: int) -> bool:
hash_map = {}
for i in range(len(nums)):
if (nums[i] in hash_map) and (i - hash_map[nums[i]] <= k):
return True
else:
hash_map[nums[i]] = i
return False
滑动窗口
维护一个长度为K+1的滑动窗口,滑动窗口中的任意两个下标差的绝对值不超过k。移动滑动窗口,如果其中有重复元素,则存在两个不同的下标 i 和 j 满足 abs(i -j) <= k且 nums[i]=nums[j]。如果所有滑动窗口中都没有重复元素,则不存在符合要求的下标。哈希表很适合判断存不存在这种问题,因为复杂度为O(1), 具体操作如下:
滑动窗口的结束下标是i, 如果i > k,则下标 i - k - 1 处的元素被移出滑动窗口;
判断 nums[i] 是否在哈希集合中,如果在哈希集合中则在同一个滑动窗口中有重复元素,返回 True,如果不在哈希集合中则将其加入哈希集合;
当遍历结束时,如果所有滑动窗口中都没有重复元素,返回 False。
时间复杂度O(n) 空间复杂度O(k)
class Solution:
def containsNearbyDuplicate(self, nums: List[int], k: int) -> bool:
hash_map = {}
for i in range(len(nums)):
if i > k:
hash_map.pop(nums[i-k-1])
if nums[i] in hash_map:
return True
else:
hash_map[nums[i]] = 1
return False