LeetCode 219. 存在重复元素 II (Contains Duplicate)[简单]

lework · 2020年05月03日 · 最后由 lework 回复于 2020年05月03日 · 130 次阅读

给定一个整数数组和一个整数 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 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

class Solution:
    def containsNearbyDuplicate(self, nums: List[int], k: int) -> bool:
        # 首先是判断是存在num[i]==num[j],存在的情况下才判断|i-j|<=k
        nums_len = len(nums)
        if nums_len <= 1:
            return False
        nums_dict = {}
        for i in range(nums_len):
            if nums[i] in nums_dict:
                if i-nums_dict[nums[i]] <= k:
                    return True
            nums_dict[nums[i]] = i
        return False
需要 登录 后方可回复, 如果你还没有账号请点击这里 注册