您的位置:首页 > 大数据 > 人工智能

[算法分析与设计] leetcode 每周一题: Contains Duplicate III

2017-11-03 02:21 549 查看
题目链接: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
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: