Leetcode: Contains Duplicate III

最新推荐文章于 2025-08-20 09:37:41 发布

YMHH

最新推荐文章于 2025-08-20 09:37:41 发布

阅读量504

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode 文章标签： leetcode 二分查找

本文链接：https://blog.youkuaiyun.com/ymhhym/article/details/47114169

Leetcode 专栏收录该内容

249 篇文章

订阅专栏

本文介绍了一种在整数数组中查找是否存在两个不同索引i和j，使得两元素值之差小于等于t且索引之差小于等于k的高效算法。通过使用STL set并结合lower_bound和upper_bound函数，实现了O(nlogk)的时间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

有些复杂，网上很多解法其实都不正确，虽然AC了。最容易想到的滑动窗口，时间复杂度O(nk)，估计会超时。二分查找的思想，可以用STL的set来实现，这样时间复杂度O(nlogk)。测试数据集很龌龊，要考虑溢出的情况。另外lower_bound和upper_bound这两个函数很容易让人迷惑。lower_bound返回大于等于target的第一个元素，upper_bound返回大于target的第一个元素。不是想象中的一个返回floor，一个返回ceiling。

class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        set<long long> counts;
        for (int i = 0; i < nums.size(); ++i) {
            if (i > k) {
                counts.erase(nums[i-k-1]);
            }
            
            auto iter = counts.lower_bound(nums[i]);
            if (iter != counts.end()) {
                // Check the bigger element
                if (*iter - nums[i] <= t) {
                    return true;
                }
                
                // Check the smaller element
                if (iter != counts.begin()) {
                    --iter;
                    if (nums[i] - *iter <= t) {
                        return true;
                    }
                }
            }
            else if (counts.size() > 0) {
               // All elements are smaller than current element
               --iter;
               if (nums[i] - *iter <= t) {
                  return true;
               }
            }

            counts.emplace(nums[i]);
        }
        
        return false;
    }
};