Leetcode 220. 存在重复元素 III C++

Leetcode 220. 存在重复元素 III

题目

在整数数组 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

题解

有序集合set，维持其大小为k，也就是说集合中最多存放的就是距离当前数为k的数。我们利用lower_bound函数二分查找，集合中大于等于当前数-t的第一个数，如果这个数，同时也满足小于等于当前数+k。显然就存在i、j满足题意了。详细过程见代码

代码

bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        set<long> list;
        int n = nums.size();
        for(int i=0; i<n; i++){
            auto s = list.lower_bound((long)nums[i]-t);		//找到第一个大于等于nums[i]-t的数
            if(s!=list.end() && *s <= (long)nums[i]+t)  return true;	//同时满足小于等于nums[i]+t，则返回true
            list.insert(nums[i]);
            if(list.size()>k){		//保证集合中的数距离当前遍历的数的距离小于等于k
                list.erase(nums[i-k]);
            }
        }
        return false;
    }

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