方法一:暴力法
维护了一个kk大小的滑动窗口来满足第一个条件,只需要通过线性搜索来检查第二个条件是否满足就可以了。超时!
当数组中的元素非常大的时候,进行数学运算可能造成溢出,用支持大数的数据类型。
class Solution {
public:
bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
for(int i=0;i<nums.size();i++){
for(int j=max(i-k,0);j<i;j++){
if(abs((double)nums[i]-(double)nums[j])<=t)
return true;
}
}
return false;
}
};
方法二:二叉搜索树
std::set::lower_bound返回一个迭代器,指向第一个关键字不小于k的元素
std::set::upper_bound返回一个迭代器,指向第一个关键字大于k的元素
假设这棵树上节点总数为 n,一个平衡树能把高度维持在 h=logn。因此这棵树上支持在O(h)=O(logn) 时间内完成插入,搜索,删除操作。
算法的伪代码:
初始化一颗空的二叉搜索树 set
对于每个元素x,遍历整个数组
在 set 上查找大于等于x的最小的数,如果s−x≤t则返回 true
在 set 上查找小于等于x的最大的数,如果x−g≤t则返回 true
在 set 中插入x
如果树的大小超过了k, 则移除最早加入树的那个数。
返回 false
我们把大于等于 x 的最小的数 s 当做 x 在 BST 中的后继节点。同样的,我们能把小于等于 x 最大的数 g 当做 x 在 BST 中的前继节点。s 和 g 这两个数是距离 x 最近的数。因此只需要检查它们和 x 的距离就能知道条件二是否满足了。
class Solution {
public:
bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
set<double> set;
for (int i = 0; i < nums.size(); ++i) {
auto s = set.lower_bound((double)nums[i] - (double)t);
if (s != set.end() && *s <= (double)nums[i] + (double)t)
return true;
set.insert(nums[i]);
if (set.size() > k) //如果树的大小超过了kk, 则移除最早加入树的那个数
set.erase(nums[i - k]);
}
return false;
}
};