DEnjoy

导航

python leetcode 日记 --Contains Duplicate II --219

题目:

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 jis at most k.

给定一个整形数组和一个整数型数k,找出在这个数组中是否存在两个相同的数,并且这两个数的下标的距离小于k。

        """
        :type nums: List[int]
        :type k: int
        :rtype: bool
        """

根据题目可知输入为一个list和一个整形,返回值为bool

因为此题不能改变数组,因此最原始的一种办法就是遍历:

 

class Solution(object):
    def containsNearbyDuplicate(self, nums, k):
        if k==0:
            return False
        while len(nums)>1:
            tem=nums.pop(0)#没那一个数出来比较,原列表少一,减少搜索空间,但减少的很慢,效率很低
            j=self.findLocation(nums,tem,k)
            if j<k and j>=0:
                return True
        return False
    def findLocation(self,L,number,k):
        for item in L:
            if item ==number:
                return L.index(item)
            k-=1
            if k==0:
                return -1
        return -1

但是在提交时发现此种方法无法通过例子list(range(30000)) 15000这个测试,分析发现因为如果数组的数全不一样,那么其复杂度为O(n2)。

在查找解决方法后,学习了python中字典的使用,参考https://leetcode.com/discuss/54123/python-concise-solution-with-dictionary

 

class Solution(object):
    def containsNearbyDuplicate(self, nums, k):
        dictionary={}
        for key,value in enumerate(nums):
            if value in dictionary and key-dictionary[value]<=k:
                return True
            dictionary[value]=key
        return False

通过使用字典,将算法复杂性变为O(n)

如果使用c++或JAVA,本题则需使用hashtable进行相似的处理。

 

posted on 2016-03-14 20:04  DEnjoy  阅读(308)  评论(0编辑  收藏  举报