leetcode算法学习(5)——存在重复元素II:判断重复元素的索引差绝对值是否至多为k

给定一个整数数组和一个整数 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

思路

• 首先判断是否存在重复元素，不存在返回False
• 遍历数组：
  • 若当前数字不在字典中，则将num：index放入字典
  • 若已存在：
    • 判断当前的索引和上一相同元素的索引差是否小于等于k，若满足，返回True。
    • 将索引更新为当前索引
• 遍历结束，返回False

代码

class Solution:
    def containsNearbyDuplicate(self, nums: List[int], k: int) -> bool:
        num_index={}
        # 没有重复元素, 返回False
        if len(set(nums))==len(nums):return False
        # 只要有一个重复元素 相差的索引号 <=k 返回True
        for index,num in enumerate(nums):
            if num in num_index and index-num_index[num]<=k:
                return True 
            num_index[num]=index
        return False