LeetCode 220 Contains Duplicate III

题目大意:给定一个数组num,问是否存在|i - j| <= k 使得 |num[i] - num[j]| <= t

本题直接用multiset做比较简便,由于本题存在一个范围限制,因此采用动态插入删除的方式维护一个大小不超过k的平衡二叉搜索树。注意每次新插入只需要考虑在数组中在其左侧长度为k-1的部分,不需要考虑右侧,因为假如右侧有满足要求的数,在之后的处理中是会被发现的。

class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        multiset<long long> window;
        for(int i = 0 ; i < nums.size(); i++)
        {
            if( i > k )
            {
                window.erase(nums[i - k - 1]);
            }
            auto pos = window.lower_bound((long long)nums[i] - t);
            if(pos != window.end() && labs(*pos - nums[i]) <= t ) return true;
            window.insert(nums[i]);
        }
        return false;
    }
};

猜你喜欢

转载自blog.csdn.net/yjr3426619/article/details/80957675