存在重复元素 III

题目描述

给定一个整数数组，判断数组中是否有两个不同的索引 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;
    }
};