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;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_36946274/article/details/81119614