LeetCode 220. 存在重复元素 III(C++)

题目:

给定一个整数数组,判断数组中是否有两个不同的索引 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

思路:

这道题与219题类似,但是需要注意此时并不是查找一个一直元素,而是查找一个范围内的元素是否存在,所以需要使用set而不是unordered_set,这里还需要了解的时lower_bound(val)返回set中大于或等于val的第一个元素位置。如果所有元素都小于val,则返回last的位置,如果lower_bound(nums[i] - t)返回值小于等于nums[i] + t,则直接返回true,否则继续遍历nums[i]
关于lower_bound(val)具体解释可参考下面的博客:
lower_bound()返回值
注意:由于算例的数字大小原因,需要使用long long替代int
错误提示

class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        set<long long> record;
        for(int i = 0; i < nums.size(); i++){
            if(record.lower_bound((long long)nums[i] - (long long)t)  != record.end() 
                   && (long long)*record.lower_bound((long long)nums[i] - (long long)t) <= (long long)nums[i] + (long long)t)
                return true;

            record.insert(nums[i]);
            if(record.size() > k)
                record.erase(nums[i - k]);
        }
        return false;
    }
};

猜你喜欢

转载自blog.csdn.net/my_clear_mind/article/details/81783907