题目描述
给定一个整数数组,判断数组中是否有两个不同的索引 i 和 j,使得 nums [i] 和 nums [j] 的差的绝对值最大为 t,并且 i 和 j 之间的差的绝对值最大为 ķ。
示例 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/contains-duplicate-iii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路
维护一个k大小的哈希表,存放下标相邻的k个数,首先保证 i 和 j 之间的差的绝对值最大为 ķ。然后利用桶排序的思想,将这k个数放进对应的桶中,桶的大小为t。这样差的绝对值最大为 t的两个数只可能在同一个桶中,或在两个相邻的桶中。
关于两个数之间的间隔,同样用到桶思想的题:最大间距
代码(c++)
class Solution {
public:
long long getID(long long i,long long t){
return i<0?(i+1)/t-1:i/t;
}
bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
if(t<0) return false;
map<long long,long long> mp;
long long tempT=(long long)t+1;
for(int i=0;i<nums.size();i++){
long long bucketID=getID(nums[i],tempT);
if(mp.find(bucketID)!=mp.end()) return true;
if(mp.find(bucketID-1)!=mp.end()&&abs(mp[bucketID-1]-nums[i])<=t) return true;
if(mp.find(bucketID+1)!=mp.end()&&abs(mp[bucketID+1]-nums[i])<=t) return true;
mp[bucketID]=nums[i];
if(i>=k) mp.erase(mp.find(getID(nums[i-k],tempT)));
}
return false;
}
};