QUESTION
medium
题目描述
给定一个整数数组,判断数组中是否有两个不同的索引 i
和 j
,使得 nums [i] 和 nums [j]
的差的绝对值最大为 t
,并且 i
和 j
之间的差的绝对值最大为 k
。
示例 1:
输入: nums = [1,2,3,1], k = 3, t = 0
输出: true
示例 2:
输入: nums = [1,0,1,1], k = 1, t = 2
输出: true
示例 3:
输入: nums = [1,5,9,1,5,9], k = 2, t = 3
输出: false
说明
SOLUTION
一开始我是想在 219
的基础上,加一层循环来保证 “nums [i] 和 nums [j]
的差的绝对值最大为 t
” 这个条件,写了如下代码:
class Solution {
public:
bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
int size = nums.size();
unordered_map<int, int> m;
for(int i = 0; i < size; i++){
for(int j = nums[i] - t; j <= nums[i] + t; j++){
if(m.find(j) != m.end()){
if(i - m[j] <= k) return true;
}
}
m[nums[i]] = i;
}
return false;
}
};
但是遇到了这个用例无法通过
[0,2147483647]
1
2147483647
在上面这种输入的情况下,这个循环 for(int j = nums[i] - t; j <= nums[i] + t; j++)
循环时间太长,导致时间超限。但是整体想法不变,加以改进,我们得到了方法一。
方法一
用有序的 map
来得到映射关系,循环的时候只判断已经形成映射关系的 key
,通过 lower_bound
找到开始位置,最后使用了 long
作为存储类型防止减法产生溢出。
class Solution {
public:
bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
int size = nums.size();
map<long, int> m;
map<long, int>::iterator it;
for(int i = 0; i < size; i++){
it = m.lower_bound((long)nums[i] - t); //nums[i] - t 可能会溢出
while(it != m.end()){
if((long)it->first - t > nums[i]) break;
if(i - it->second <= k) return true;
it++;
}
m[nums[i]] = i;
}
return false;
}
};
方法二
这个方法是从别的博客看来的,可能有点难理解。
这里我们使用map数据结构来解,用来记录数字和其下标之间的映射。 这里需要两个指针i和j,刚开始i和j都指向0,然后i开始向右走遍历数组,如果i和j之差大于k,且m中有nums[j],则删除并j加一。这样保证了m中所有的数的下标之差都不大于k,然后我们用map数据结构的lower_bound()函数来找一个特定范围,就是大于或等于nums[i] - t的地方,所有小于这个阈值的数和nums[i]的差的绝对值会大于t (可自行带数检验)。然后检测后面的所有的数字,如果数的差的绝对值小于等于t,则返回true。最后遍历完整个数组返回false。代码如下:
class Solution {
public:
bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
map<long long, int> m;
int j = 0;
for (int i = 0; i < nums.size(); ++i) {
if (i - j > k) m.erase(nums[j++]);
auto a = m.lower_bound((long long)nums[i] - t);
if (a != m.end() && abs(a->first - nums[i]) <= t) return true;
m[nums[i]] = i;
}
return false;
}
};