220.存在重复元素 III (Medium)*

题目描述*

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

示例*

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

输出: true

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

输出: true

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

输出: false

代码*

题目的意思就是在任意有效区间 [i, i + k] 中是否存在两个数之差小于等于 t。

暴力方法是枚举所有相差不超过 k 的下标对，判断是否满足条件，毫无意外地超时了。

可以用滑动窗口的思想，维护一个结点总数为 k 的二叉搜索树，遍历 nums，找到 nums[i] 在二叉搜索树中的前驱和后继结点，如果满足相差小于等于 t，则返回 true。

abs(nums[i] - nums[j]) <= t <==> nums[i] - t <= nums[j] <= nums[i] + t

暴力二叉搜索树

class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        for(int i = 0; i < nums.size(); i++) {
            for(int j = i + 1; j < nums.size() && j <= i + k; j++) {
                if(abs(static_cast<long>(nums[i]) - static_cast<long>(nums[j])) <= t) {
                    return true;
                }
            }
        }
        return false;
    }
};

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

最后更新: July 23, 2022