LeetCode-220. 存在重复元素 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

思路：桶排序

1. 定义桶的大小是t+1, nums[i]/(t+1)决定放入几号桶,这样在一个桶里面的任意两个的绝对值差值都是<=t的
2. 遍历nums每一个元素，并把他们放入相应的桶中，下面两种情况会返回true：

• 要放入的桶中已经有其他元素了，这时将nums[i]放进去满足差值<=t
• 存在前面一个桶的元素并且与nums[i]的差值<=t或者 存在后面一个桶的元素并且与nums[i]的差值<=t，前后桶的元素最多也只能有一个。

3. 考虑限制条件：桶中的索引差最大为k 当i>=k时：删除存放在nums[i-k]的那个桶(编号为nums[i-k]/(t+1))，这样就能保证遍历到第i+1个元素时，全部桶中元素的索引最小值是i-k+1

代码实现：

class Solution {
 
  public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
       if (t < 0) {
           return false;
       }
       HashMap<Long, Long> map = new HashMap<>();
       int n = nums.length;
       for (int i = 0; i < n; i++) {
           //删除窗口中第一个数字
           if (i >= k) {
               map.remove(getId(nums[i - k - 1], t+1));
           }
           //得到当前数的桶编号
           long id = getId(nums[i], t+1);
           //要放入的桶中已经有其他元素了
           if (map.containsKey(id)) {
               return true;
           }
           //存在后面一个桶的元素并且与nums[i]的差值<=t
           if (map.containsKey(id + 1) && map.get(id + 1) - nums[i] <= t) {
               return true;
           }
           //存在前面一个桶的元素并且与nums[i]的差值<=t 
           if (map.containsKey(id - 1) && nums[i] - map.get(id - 1) <= t) {
               return true;
           }
           map.put(id, (long) nums[i]);
       }
       return false;
  }

   private long getId(long num, long w) {
       if (num >= 0) {
           return num / w;// 一个桶里边数字范围的个数是 t + 1
       } else {
           // 在Java中, -3 / 5 = 0 a但我们需要的是 -3 / 5 = -1.
           return (num + 1) / w - 1;
       }
   }
}

复杂度分析

• 时间复杂度：$O(n)$
• 空间复杂度：$O(\min (n,k))$
  需要开辟的额外空间取决了散列表的大小，其大小跟它所包含的元素数量成线性关系。散列表的大小的上限同时由 $n$ 和 $k$ 决定。因此，空间复杂度为 $O(\min (n,k))$。