题目大意:给定一个数组num,问是否存在|i - j| <= k 使得 |num[i] - num[j]| <= t
本题直接用multiset做比较简便,由于本题存在一个范围限制,因此采用动态插入删除的方式维护一个大小不超过k的平衡二叉搜索树。注意每次新插入只需要考虑在数组中在其左侧长度为k-1的部分,不需要考虑右侧,因为假如右侧有满足要求的数,在之后的处理中是会被发现的。
class Solution {
public:
bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
multiset<long long> window;
for(int i = 0 ; i < nums.size(); i++)
{
if( i > k )
{
window.erase(nums[i - k - 1]);
}
auto pos = window.lower_bound((long long)nums[i] - t);
if(pos != window.end() && labs(*pos - nums[i]) <= t ) return true;
window.insert(nums[i]);
}
return false;
}
};