算法学习之二分搜索树底层实现的顺序性

Jackey C/C++ 497 次浏览 没有评论

题目要求:

给你一个整数数组 nums 和两个整数 k 和 t 。请你判断是否存在 两个不同下标 i 和 j,使得 abs(nums[i] - nums[j]) <= t ,同时又满足 abs(i - j) <= k 。

如果存在则返回 true,不存在返回 false。

 

示例 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
提示:

0 <= nums.length <= 2 * 104
-231 <= nums[i] <= 231 - 1
0 <= k <= 104
0 <= t <= 231 - 1

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/contains-duplicate-iii

解题代码:

// 时间复杂度: O(nlogn)
// 空间复杂度: O(k)
class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        set<long long> record; // 防止32位整型溢出
        for (int i = 0; i < nums.size(); ++i) {
            if (record.lower_bound((long long)nums[i] - (long long)t) != record.end() &&
            *record.lower_bound((long long)nums[i] - (long long)t) <= (long long)nums[i] + (long long)t)
                return true;

            record.insert(nums[i]);

            // 保证record中最多有k个元素
            if (record.size() == k + 1)
                record.erase(nums[i-k]);
        }
        return false;
    }
};

 

发表评论

您的电子邮箱地址不会被公开。

Go