Contains Duplicate III 存在重复元素 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

思路:我们需要维护一个长度为k的滑动窗口,然后在滑动窗口里寻找满足条件的等式:

-t<=|nums[i]-x|<=t  =>  x>=nums[i]-t && x<=(nums[i]+t)

这里的x表示滑动窗口中存在的数字,就是我们需要找的数字,那么只要找到这个数字就可以了,我们可以调用自带排序功能的set或者map的lower_bound函数,找到下标后观察是否上限满足要求,如果满足直接返回true即可。

参考代码:

class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
	set<int> window;
	for (int i = 0; i < nums.size(); i++) {
		if (i > k) window.erase(nums[i-k-1]); 
		long long lower =((long long)nums[i]- (long long)t)<INT_MIN?INT_MIN:(nums[i] - t) ; //  -t<=|nums[i]-x|<=t  =>  x>=nums[i]-t && x<=(nums[i]+t)
		auto pos = window.lower_bound(lower);
		long long uper = ((long long)nums[i] + (long long)t) > INT_MAX ? INT_MAX : (nums[i] + t);
		if (pos != window.end() && *pos <= uper) return true;
		window.insert(nums[i]);
	}
	return false;       
    }
};

猜你喜欢

转载自blog.csdn.net/qq_26410101/article/details/83051204