leetcode 220.存在重复元素III

leetcode 220.存在重复元素III

题干

给你一个整数数组 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

知识点&算法

寻找区间内值在一定范围内的两个元素是否存在，很容易想到 滑动窗口 + 内部有序数据结构 的做法，但在实现的过程中会遇到两个问题：
1.c++常用的几个内部有序的数据结构(set,multiset等)的erase方法在传入值时会将所有符合的元素删去。
2.即使滑动窗口的内部元素已经有序，如何搜索符合条件的元素？
对于问题1，可以发现，在维护滑窗的过程中，如果一个值不符合要求，可能是值不符合要求，也可能是下标不符合要求。对于值不符合要求的情况，所有等值的元素都是等价的，所以全部删去不影响后续的统计；
对于下标不符合要求的情况，只要是滑窗内部最靠右的元素不符合要求，那么所有左侧的元素也都是无效的，可以删去。综上可以得到策略，在滑窗内部寻找值符合要求的下标最靠右的元素。然而实际上在有序数
据结构的内部相等的值都是挨在一起的，所以全部删去实际上不影响答案。
对于问题2，既然元素已经有序，很容易想到二分法，那么利用STL的成员函数来搜索值满足条件的元素即可。

题解

class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        set<int> w;
        int n = nums.size();
        if(n == 0 || n == 1) return false;
        int l = 0,r = 0;
        while(r < n){
            if(!w.empty()){
                auto itr = w.end();
				//避免整型极限数据溢出
                if(nums[r] < INT_MIN + t) itr = w.lower_bound(INT_MIN);
                else itr = w.lower_bound(nums[r] - t);
                if(itr != w.end() && (*itr <= nums[r] || *itr <= nums[r] + t)) return true;
            }
            if(r - l >= k){
                w.insert(nums[r]);
                w.erase(nums[l]);
                l++;
            }else{
                w.insert(nums[r]);
            }
            r++;
        }
        return false;
    }
};