滑动窗口+二叉搜索树
这题跟前两题不一样,已经跟重复数字没关系了
对于数组内的某一个整数,如果能够求出大于它的最小整数那么我们就可以判断差值是否小于等于t。能方便的求出最大下限或最小上限,最先想到的数据结构是二叉搜索树BST,因为左孩子就是最大下限,右孩子就是最小上限。
可以用一个大小为k的滑动窗口,将窗口内的元素组织成一个BST,每次向前滑动一步,添加一个新元素,同时删除一个最老的元素,如此不断向前滑动,不断更新BST。如果当前整数和BST中大于它的最小整数的差的绝对值小于等于t是 ,则返回true,如果直到扫描完数组,都没有符合条件的,则返回false。
对于BST数据结构,可以用现成的,C++里是 multiset
注意一个坑,测试数据里有超过int的数据 难怪这题差评那么多了: )
class Solution
{
public:
bool containsNearbyAlmostDuplicate(vector<int> &nums, int k, long t)
{
multiset<long> bst;
for (int i = 0; i < nums.size(); i++)
{
if (i > k)
bst.erase(bst.find(nums[i - k - 1]));
auto it = bst.lower_bound(nums[i] - t);
if (it != bst.end() && abs(nums[i] - *it) <= t)
return true;
bst.insert(nums[i]);
}
return false;
}
};