题目:
Given an array of integers, find out whether there are two distinct indices i and j in the array such that the absolute difference between nums[i] and nums[j] is at most t and the absolute difference between i and j is at most k.
运行结果:
思路解析:
这题要求|nums[i]-nums[j]|≤t, 同时保证| i-j | <=k。可以维护一个窗口,大小不超过k,滑动窗口比较nums的值是否满足要求。故可以采用set来存储窗口的值。使用两个指针i 和 j,初始时i和j 同时指向0 的位置,然后指针i向右滑动,当i-j的值超过k时,则移动 j,则可以保证窗口大小不超过K 。set中存储i和 j 范围内nums的值,由于窗口限制保证了| i-j | <=k的条件,故只需考虑set中是否存在两个值的差的绝对值小于等于t。可以采用set的lower_bound函数(返回set中第一个大于等于查找值的迭代器)。由于需要满足|nums[i]-nums[j]|≤t,则可得nums[j]≥nums[i]-t且nums[j]≤nums[i]+t
,故只需保证set中存在数值在范围nums[i]-t到nums[i]+t内即可。通过采用set的lower_bound函数即可满足第一个条件,然后判断第二个条件即可。
代码:
class Solution {
public:
bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
set<long long> s;
for(int i=0,j=0;i<nums.size();i++){
if(i-j>k){
s.erase(nums[j++]);
}
auto it=s.lower_bound((long long)nums[i]-t);
if(it!=s.end()&&abs(nums[i]-*it)<=t) return true;
s.insert(nums[i]);
}
return false;
}
};