题目链接:https://leetcode.com/problems/contains-duplicate-iii/description/
题目大意: 给定整数数组num,找出数组中是否存在两个不同的索引i和j,使得num [i]和nums [j]之间的绝对差值最多为t,i和j之间的绝对差值最多为ķ。
思路:
一开始我是用暴力求解,遍历数组与其后至多8个元素比较,从而判断是否存在满足关系,然而超时了,一时间想不到解决方法,于是查看了其他博客,发现了一个比较简单又巧妙的方法:维护一个数据池(初始为空),遍历该数组,如果遍历的元素num[i] 无法在池中找到 n (n >= | num[i] - t |)则将该元素放入池中,并且进入下一轮迭代。如果遍历完毕,还未找到相关的n,则表示该数组不能满足关系。此外,该数据池不可以大于k,保证了i - j = k. 代码如下:
class Solution {
public:
bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
multiset<long long> dataset;
for(int i = 0; i < nums.size(); i++) {
if(dataset.size() > k) dataset.erase(nums[i - k - 1]);
auto it = dataset.lower_bound((long long)nums[i] - t); // note that it will overflow
if(it != dataset.end() && abs(*it - nums[i]) <= t) return true;
dataset.insert(nums[i]);
}
return false;
}
};
注意:此题的测试数据有些十分极致大,进行+-操作会溢出,因此需要long long