1. 问题描述
给定一个整数数组nums[],查找是否存在两个下标i和j,满足|numsi?numsj|≤t 且 |i?j|≤k。
2. 方法与思路
总得思路就是:“滑动窗口”+unordered_map。
推理过程如下:
|numsi?numsj|≤t?|numsi/t?numsj/t|≤1;
由上式可以推出:|?numsi/t???numsj/t?|≤1
等价于:?numsi/t?∈{?numsj/t?,?numsj/t??1,?numsj/t?+1}。
我们只需要维持一个大小为K的窗口,键值为?numsi/t?, 值为numsi。然后循环遍历nums[]数组,与unordered_ map中键集对应的值进行差得绝对值运算,再判断即可。
对于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;
}
};
时间: 2024-12-28 21:03:32