class Solution {
public:
bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
int n=nums.size();
multiset<long> window;
if(n==0 || n==1) return false;
if(k==0) return false;
int start=0,end=1;
int setsize=1;
window.insert((long)nums[0]);
while(end<n){
if(window.lower_bound(nums[end])!=window.end() && abs(*window.lower_bound(nums[end])-nums[end])<=t) return true;
multiset<long>::iterator it=window.upper_bound(nums[end]);
if(it!=window.begin()){
it--;
if(abs(*it-nums[end])<=t) return true;
}
if(setsize==k){
multiset<long>::iterator it=window.find(nums[start]);
window.erase(it);
start++;
setsize--;
}
window.insert((long)nums[end]);
setsize++;
end++;
}
return false;
}
};
快还是挺快的,内存消耗挺大。
本质还是滑动窗口,窗口里放k个数,然后第k+1个数找一下窗口里有没有和自己绝对值之差小于t的,有则查找成功返回true。没有的话把这个数插入窗口,然后把最早放进窗口的数删掉。
窗口内为了加快查找速度用了multiset。set和map都是用红黑树实现的,查找效率在logN。
之前和同学吐槽为什么set有upperbound(大于key的最小值)和lowerbound(大于等于key的最小值)却不提供小于key的最大值。做这题突然明白了,蠢到自己了。因为set内部是从小到大有序的,找到大于等于key的最小值,再给iterator--就是小于key的最大值了……
leetcode上还介绍了一种桶排序+哈希的方法,效率也差不多。