leetcode 220. 存在重复元素 III

如果直接用遍历算法,那就是来个二重循环,每一步都计算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;
    }
};
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值