https://leetcode.com/problems/contains-duplicate-ii/
对于array 的题目,一定要想到 sort, binary search, 当要改进 O(N2) 时,要想到hash map。
而对于hash map,有的时候是scan一遍之后建立完hash map再作其余判断。而像这道题,是可以在建立hash map的过程中,进行判断的。
重点关注这一类问题
因为这里要判断
j−i<=k
, 我们只需要使得 hash map的value保存最近的index就行。
见code:
class Solution(object):
def containsNearbyDuplicate(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: bool
"""
mydict = {}
for i in xrange(len(nums)):
if nums[i] in mydict:
if i - mydict[nums[i]][0] <= k:
return True
else:
mydict[nums[i]][0] = i
else:
mydict[nums[i]] = [i]
return False