219 存在重复元素 II(字典)

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

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值