220. 存在重复元素 III

题目描述

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/contains-duplicate-iii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

存下K个最近访问的元素，遍历后续的元素，从保存的K个找元素值最相近的，lower_bound迭代器的前后，看是否满足条件，把超出窗口K的元素剔除。

代码

class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        if (k <= 0) return false;
        set<long> s;
        int e_index = 0;
        for (int i = 0; i < nums.size(); ++i)
        {
            // 最近的元素，也可能是它前面一个元素
            auto iter = s.lower_bound(nums[i]);
            if (iter != s.end() && abs(*iter - nums[i]) <= t) return true;
            if (iter != s.begin() && abs(*--iter - nums[i]) <= t) return true;
            if (i >= k) s.erase(nums[e_index++]);
            s.insert(nums[i]);
        }
        return false;
    }
};