题目:
给定一个整数数组和一个整数 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
参考解答:
https://blog.youkuaiyun.com/qq_34364995/article/details/80640458
思路:
1.使用字典d来存储,以数组元素为键,索引为值,遍历数组。
2.若nums[j]存在于字典中,则转入3,否则添加键值对d[nums[j]] = j。
3.进行索引值判断,若索引差值绝对值大于k,将字典d[nums[j]] = j进行更新,然后继续遍历数组返回1。若索引差值绝对值小于k,则return True。
4.return False。
class Solution(object):
def containsNearbyDuplicate(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: bool
"""
d = {}
for i in range(len(nums)):
if nums[i] in d:
if -k <= i - d[nums[i]] <= k:
return True
else:
d[nums[i]] = i
d[nums[i]] = i
return False
参考解答2:
https://blog.youkuaiyun.com/qq_17550379/article/details/80638690
我们通过建立一个record
查找表,表中存的是窗口中的数,另外我们要注意的是,当窗口的大小>k
的时候,我们要移除record
中最左边的元素(保证我们窗口中有<=k
个数)
class Solution:
def containsNearbyDuplicate(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: bool
"""
record = set()
for i, num in enumerate(nums):
if num in record:
return True
record.add(num)
if len(record) == k + 1:
record.remove(nums[i - k])
return False
我们分析这个算法的时间复杂度,这个算法的时间复杂度是O(n)
这个级别的。如果我们查找表选用dict
,有该怎么做呢?我们的key
是nums
中的元素,而value
是对应元素在nums
中的位置。
class Solution:
def containsNearbyDuplicate(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: bool
"""
record = {}
for i, num in enumerate(nums):
if num in record and i - record[num] <= k:
return True
record[num] = i
return False