LeetCode算法——219.存在重复的数2

https://leetcode.cn/problems/contains-duplicate-ii/description/?envType=study-plan-v2&envId=top-interview-150

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

提示：

1 <= nums.length <= 105
-109 <= nums[i] <= 109
0 <= k <= 105

# 建立一个字典，key是数字，value的下标组成的list，然后再遍历这个字典，判断前后项是否超过K

class Solution:
    def containsNearbyDuplicate(self, nums: List[int], k: int) -> bool:

        map_ = {}

        cnt = 0
        for each in nums:
            if each not in map_.keys():
                map_[each] = [ cnt ]
            else:
                map_[each].append( cnt )

            cnt = cnt + 1

        print(map_)

        for key in map_.keys():
            for i in range(len(map_[key])-1):
                if (map_[key][i+1] - map_[key][i] <= k ):
                    return True
        return False