leetcode 220. 存在重复元素 III(C++)

给定一个整数数组,判断数组中是否有两个不同的索引 i 和 j,使得 nums [i] 和 nums [j] 的差的绝对值最大为 t,并且 i 和 j 之间的差的绝对值最大为 ķ

示例 1:

输入: nums = [1,2,3,1], k = 3, t = 0
输出: true

示例 2:

输入: nums = [1,0,1,1], k = 1, t = 2
输出: true

示例 3:

输入: nums = [1,5,9,1,5,9], k = 2, t = 3
输出: false

思路:使用set维护一个窗口,当能从这个窗口中找到介于nums[i]-t 和nums[i]+t的数时,说明存在。使用lower_bound找到窗口中大于等于nums[i]-t的那个数tmp,只要这个tmp满足条件即可。

C++

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

猜你喜欢

转载自blog.csdn.net/qq_27060423/article/details/93347163