219. Contains Duplicate II [easy] (Python)

来源：互联网发布：vue.js深入浅出pdf 编辑：程序博客网时间：2024/05/23 23:25

题目链接

https://leetcode.com/problems/contains-duplicate-ii/

题目原文

Given an array of integers and an integer k, find out whether there are two distinct indices i and j in the array such that nums[i] = nums[j] and the difference between i and j is at most k.

题目翻译

给定一个数组，和一个整数k，判断该数组中是否存在不同下标的 i 和 j 两个元素，使得 nums[i] = nums[j]，且 i 和 j 的差不超过k。

思路方法

暴力两重循环不可取，无法AC。。。

思路一

遍历所有元素，将元素值当做键、元素下标当做值，存放在一个字典中。遍历的时候，如果发现重复元素，则比较其下标的差值是否小于k，如果小于则可直接返回True，否则更新字典中该键的值为新的下标。

代码

class Solution(object):    def containsNearbyDuplicate(self, nums, k):        """        :type nums: List[int]        :type k: int        :rtype: bool        """        num_map = {}        for i in xrange(len(nums)):            if nums[i] in num_map and i - num_map[nums[i]] <= k:                return True            else:                num_map[nums[i]] = i        return False

思路二

在遍历列表的时候维护一个集合，集合中保存当前元素前面的k个元素，每次访问一个元素时判断是否在该集合中出现过。

代码

class Solution(object):    def containsNearbyDuplicate(self, nums, k):        """        :type nums: List[int]        :type k: int        :rtype: bool        """        window = set([])        for i in xrange(len(nums)):            if i > k:                window.discard(nums[i-k-1])            if nums[i] in window:                return True            else:                window.add(nums[i])        return False

PS: 新手刷LeetCode，新手写博客，写错了或者写的不清楚还请帮忙指出，谢谢！
转载请注明：http://blog.csdn.net/coder_orz/article/details/51674266

0 0