1. 问题描述:
给定一个整数数组和一个整数 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/contains-duplicate-ii
2. 思路分析:
① 一个比较容易想到的思路是我们可以使用字典来将相同的数字对应的下标放在一起,对于python语言来说可以使用collections.defaultdict(list)方法来创建字典,键为元素值,值为元素值对应的下标,循环遍历nums数组当发现当前的元素值不在字典中的时候那么字典会自动创建一个新的list,我们只需要将这个元素添加到字典中即可,当发现当前的元素值存在于字典中的时候那么就需要比对当前的元素值与字典中上一个相同的元素的下标的绝对值是否小于等于k:相邻的两个元素的距离肯定是最小的,比较之后就可以判断当前是否存在这样的元素对。除了上面的思路之外其实我们在遍历的时候只需要维护最近的那个元素的位置即可,最近元素那么比较距离的时候肯定是比较短的,所以也可以在字典中维护一个最近元素的下标即可。
② 除了上面的这种方法,领扣的官方提供的是借助于set集合来实现的,遍历数组,当发现元素值存在的时候那么直接返回True,不存在set集合中那么需要将其加入到set集合中,当发现set集合中的元素超过7的时候那么就需要删除set中最开始的那个元素,这个思路也是挺好理解的
3. 代码如下:
字典:
import collections
from typing import List
class Solution:
def containsNearbyDuplicate(self, nums: List[int], k: int) -> bool:
dic = collections.defaultdict(list)
for key, value in enumerate(nums):
# 值不存在于字典中
if value in dic:
# 比较字典中最近的一个元素的索引是否相差是否小于等于k, 列表的负数索引表示的是倒数开始计算起
if abs(dic[value][-1] - key) <= k:
return True
dic[value].append(key)
return False
import collections
from typing import List
class Solution:
def containsNearbyDuplicate(self, nums: List[int], k: int) -> bool:
dic = collections.defaultdict(int)
for i in range(len(nums)):
if nums[i] in dic:
if abs(i - dic[nums[i]]) <= k: return True
# 只维护最新元素的下标
dic[nums[i]] = i
return False
set集合:
import collections
from typing import List
class Solution:
def containsNearbyDuplicate(self, nums: List[int], k: int) -> bool:
rec = set()
for key, value in enumerate(nums):
if value in rec: return True
rec.add(value)
if len(rec) > k: rec.remove(nums[key - k])
return False