Leetcode 220. Contains Duplicate III

2018-01-12 14:45:56 浏览数 (2)

Given an array of integers, find out whether there are two distinct indices i and j in the array such that the absolute difference between nums[i] and nums[j] is at most t and the absolute difference between i and j is at most k.

219的再升级版,不但要求下标差的绝对值不超过k,还要求数值差的绝对值不超过t。

最外层遍历数组,对于每一个位置,我们需要知道它前k个位置中和它在数值上最接近的位置,显然要用到搜索树。

不需要手写,因为set就是搜索树的一种实现。

代码语言:javascript复制
class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, long long t) {
        set<long long> s;
        for(int i = 0; i < nums.size(); i  )
        {
            if(i > k) s.erase(nums[i - k - 1]);//删除前面第k 1个元素
            set<long long>::iterator pos = s.lower_bound(nums[i] - t);//如果*pos >= nums[i] - t,则*pos - nums[i] >= -t
            if(pos != s.end() && *pos - nums[i] <= t) return true; //存在*pos,且满足上界条件
            s.insert(nums[i]);
        }
        return false;
    }
};

0 人点赞