LeetCode - 220. 存在重复元素 III

掉坑总结：

1）int类型值与参数t相加的时候可能出现溢出，编码的时候没有考虑到；

2）理解为什么不能用upper_bound来判断，显然如果集合中存在的元素均远远大于当前给定的比较元素num，且num+t也小于给定的元素，则不管upper_bound(num-t) 或者 upper_bound(num+t)均指向第一个元素，无法判断。

描述

在整数数组 nums 中，是否存在两个下标 i 和 j，使得 nums [i] 和 nums [j] 的差的绝对值小于等于 t ，且满足 i 和 j 的差的绝对值也小于等于 ķ 。

如果存在则返回 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/contains-duplicate-iii/
 

求解

    class Solution {
    public:
        // 方法一
        bool containsNearbyAlmostDuplicate(vector<int> &nums, int k, int t) {
            const int n = nums.size();
            std::set<long long> record;
            for (int i = 0; i < n; ++i) {
                long long num = nums[i];
                const auto lit = record.lower_bound(num - t);
                // 注意理解该判断，[1,5,9,1,5,9],k=2,t=3
                // 当i = 3的时候, nums[3] = 1, record中有两个元素[5,9]
                // 这个时候在record中搜索1-3=-2, 1+3=4的 lower_bound / upper_bound 都能搜索到
                // 然而，lower_bound(-2)指向5，比4还大，显然不合理
                if (lit != record.end() && (*lit <= num + t)) {
                    return true;
                }
                record.emplace(num);
                if (record.size() == k + 1) {
                    record.erase(nums[i - k]);
                }
            }
            return false;
        }

        // 方法二，错误，不能用upper_bound来进行判断
        bool containsNearbyAlmostDuplicate_ERROR(vector<int> &nums, int k, int t) {
            const int n = nums.size();
            std::set<long long> record;
            for (int i = 0; i < n; ++i) {
                long long num = nums[i];
                const auto uit = record.upper_bound(num - t);
                if (uit != record.end() && (*uit >= num - t)) {
                    return true;
                }
                record.emplace(num);
                if (record.size() == k + 1) {
                    record.erase(nums[i - k]);
                }
            }
            return false;
        }

    };