给定一个整数数组nums[],查找是否存在两个下标i和j,满足
总得思路就是:“滑动窗口”+unordered_map。
推理过程如下:
由上式可以推出:
等价于:
我们只需要维持一个大小为K的窗口,键值为
对于STL中关于unordered_map的相关可参考官方References 对unordered_map讲解。
注意: unordered_map是一种特殊的hash map,它不按照键值进行排序。通过hash表的算法,查找效率为O(1)。但会消耗一定内存。进行差值运算时需转换为long类型,否则会溢出。
class Solution {
public:
bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
if(t < 0 | k <1)
return false;
int i,key;
unordered_map<int,int> dict;
for(i =0; i < nums.size(); i++)
{
key = nums[i]/max(1,t);
//map<int,int>::iterator it;
if( (dict.find(key) != dict.end() && abs(nums[i] - dict[key]) <= t) ||
(dict.find(key-1) != dict.end() && abs((long)nums[i] - (long)dict[key-1]) <= t) ||
(dict.find(key+1) !=dict.end() && abs(nums[i] - dict[key+1]) <= t))
{
return true;
}
//dict[key] = nums[i];
dict.insert(pair<int,int>(key,nums[i]));
if(i >= k)
{
dict.erase(nums[i-k]/max(1,t)); //删除窗口大小之外的键值
}
}
return false;
}
};
Leetcode 220 Contains Duplicate III
原文地址:http://blog.csdn.net/jeanphorn/article/details/46625729