一.解法
https://leetcode-cn.com/problems/contains-duplicate-ii/
要點:hashset,滑動窗口
和218題類似,但要注意相鄰相同數的位置差大小是否大于k。
有兩種方法,java和c++用了滑動窗口,保證hashset的大小小于等于k。
Python用了另一種方法,用字典哈希表,hashmap[nums[i]]=i來存放最新一次nums[i]這個元素出現的位置,如果新的i減去hashmap[nums[i]]大于k,就更新hashmap[nums[i]]=i,否則返回true。
二.Python實現
class Solution:
def containsNearbyDuplicate(self, nums: List[int], k: int) -> bool:
hashset={}
length=len(nums)
for i in range(0,length):
if hashset.get(nums[i])==None:
hashset[nums[i]]=i
else:
if i-hashset[nums[i]]<=k:
return True
hashset[nums[i]]=i
return False
三.C++實現
class Solution {
public:
bool containsNearbyDuplicate(vector<int>& nums, int k) {
unordered_set<int> set; //搜索、插入和移除平均常數時間復雜度,不會超時
for(int i = 0; i < nums.size(); i++)
{
if(set.find(nums[i])!= set.end())
return true;
set.insert(nums[i]);
if(set.size() > k )
set.erase(nums[i-k]); //滑動窗口長度最大為k
}
return false;
}
};
四.java實現
class Solution {
public boolean containsNearbyDuplicate(int[] nums, int k) {
Set<Integer> set = new HashSet<>();
for (int i = 0; i < nums.length; ++i) {
if (set.contains(nums[i])) return true;
set.add(nums[i]);
if (set.size() > k) {
set.remove(nums[i - k]);
}
}
return false;
}
}