LeetCode 存在重复元素 III

最新推荐文章于 2021-11-24 17:03:43 发布

转载最新推荐文章于 2021-11-24 17:03:43 发布 · 218 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：https://blog.youkuaiyun.com/qq_36946274/article/details/81119614

文章标签：

#leetcode

examples 专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一种使用滑动窗口结合二叉搜索树的数据结构解决特定算法问题的方法。通过维护一个大小固定的二叉搜索树，实现对数组中特定条件元素的高效查找。适用于需要快速判断区间内元素是否满足一定条件的场景。

数学描述一下，是否存在i,j使得：

$|i-j|<=k; |nums[i]-nums[j]|<=t$ 成立。

我是在二叉搜索树里看到这个题目的，所以应该用二叉搜索树来解决。

解决思路：维护一个大小为k的二叉搜索树，该树的元素为一个在数组上大小为k+1的滑动窗口所包括的元素，该窗口从左往右滑动。滑动接触到新的元素nums[i]，此时应检查二叉树有无元素x满足：

$-t\leq x-nums[i]\leq t$

拆解成：

$x\geq nums[i]-t$ ①

$x-nums[i]\leq t$ ②

如果存在这样的x，那么便可返回true。

寻找①中最小的x，如果还不满足②，那么当前窗口里没有符合条件的x，将新元素插入二叉搜索树，并继续滑动。

STL里set的lower_bound(n)可以返回第一个大于等于n的元素，于是可以用这个函数来求最小的x。

将上述思路整理成代码即可。

class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {        
        //滑动窗口
		//由set构成的二叉搜索平衡树（红黑树）
        //long用于LeetCode的测试用例。。
		set<long> buffer;
		long lt = t;
		for (int i = 0; i < nums.size(); i++)
		{
			if (i > k)
				buffer.erase(nums[i - k-1]);
			//lower_bound找到第一个大于等于的元素
			set<long>::iterator it = buffer.lower_bound(nums[i]-lt);
			//满足不等式
			if (it != buffer.end() && (*it)-nums[i] <= lt)
				return true;


			buffer.insert(nums[i]);

		}
        
        
        return false;
}
};

原文地址：

https://blog.youkuaiyun.com/qq_36946274/article/details/81119614