您的位置:首页 > 其它

leetcode 220. 存在重复元素 III

2018-07-19 17:31 281 查看

如果直接用遍历算法,那就是来个二重循环,每一步都计算k次,复杂度为O(N*K),很机械,很愚蠢。用这种傻子都能想到的算法,不配做程序员。
我们略微分析一下上面的傻子算法,在遍历K个范围内的数的时候,如果明显遇到了一个比上一个数还大的数字,我们还计算它干嘛呢?也就是说,这K个数里面,我们只跟离t最近的数求绝对值不就行了吗?
我们只需要用set维护一个长度为k的窗口,什么?谁说窗口只能用双指针维护了?用set正合适,它底层使用红黑树实现的。这样我们在遍历的过程中,每次都找到这个窗口中的离t最近的值求差就行了,O(N*K)降为了O(N*log K)。
什么?如何找离t最近的值?看下面的代码!

class Solution {
public:
bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
//
set<long> ss;
long lt=t;
for(int i=0;i<nums.size();i++){
if(i>k) ss.erase(nums[i-k-1]);
set<long>::iterator it=ss.lower_bound(nums[i]-lt);
if(it!=ss.end() && (*it)-nums[i]<=lt) return true;
ss.insert(nums[i]);
}
return false;
}
};
阅读更多
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐