leetcode挑战5新手讲解向，220. 存在重复元素 III

原题

给你一个整数数组 nums 和两个整数 k 和 t 。请你判断是否存在 两个不同下标 i 和 j，使得 abs(nums[i] - nums[j]) <= t ，同时又满足 abs(i - j) <= k 。
如果存在则返回 true，不存在返回 false。
Given an integer array nums and two integers k and t, return true if there are two distinct indices i and j in the array such that abs(nums[i] - nums[j]) <= t and abs(i - j) <= k.

Example 1:

Input: nums = [1,2,3,1], k = 3, t = 0
Output: true
Example 2:

Input: nums = [1,0,1,1], k = 1, t = 2
Output: true
Example 3:

Input: nums = [1,5,9,1,5,9], k = 2, t = 3
Output: false
 

Constraints:

0 <= nums.length <= 2 * 104
-231 <= nums[i] <= 231 - 1
0 <= k <= 104
0 <= t <= 231 - 1

答案

class Solution {
    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
       int n = nums.length;
        TreeSet<Long> ts = new TreeSet<>();
        for (int i = 0; i < n; i++) {
            Long u = nums[i] * 1L;
            // 从 树中拿出小于等于u的最接近u的数
            Long l = ts.floor(u); 
            // 从 树中拿出大于等于 u 的最接近 u 的数
            Long r = ts.ceiling(u); 
            if(l != null && u - l <= t) return true;
            if(r != null && r - u <= t) return true;
            // 将当前数加到树中排序，并移除下标范围不在 [max(0, i - k), i) 的数（维持滑动窗口大小为 k）
            ts.add(u);
            if (i >= k) 
            	ts.remove(nums[i - k] * 1L);
        }
        return false;
    }
}

解析

1. treeSet数据结构是红黑树，也即自平衡的排序二叉树，数据按照大小排序在树中，插入数据的时间复杂度logn，查找的时间复杂度logn，所以该答案的算法的时间复杂度为nlogn
2. 遍历所有数据，每次遍历都将数据存放到树中，存放的同时将超出范围的值从树中去除
3. 每次遍历时还要判断当前范围内是否已经有有效值出现了
4. 利用红黑树的的特性，存放到红黑树中的数据已经做好了排序了，所以可以快速的取出红黑树中在某个范围内的值
5. 取出距离当前值最近的两个数（因为新加的数才可能会出现有效值，先前的数字已经全部计算过了），看是否符合条件，符合条件即可得到结果

思考

1. 抽象化该问题，其实就是要判断数组中，是否存在这么一个范围（k），在这个范围里面，有那么两个数字，他们的差值非常靠近（t）
2. 所以问题很简单，首先就是要找出所有符合k的范围，其实就是一个遍历放到tree中，tree中的个数超出（k）后就移除最前面的那一个
3. 接着就是找出范围内的有效值了，如果一个个匹配的话那么速度未免也太慢了，使用排序算法排完了序后再去找目标值就快很多了，所以这里就用了treeSet的结构，利用红黑树可以快速的完成排序以及查找。
4. 查找目标值的时候这里也很巧妙，只查找了一次就得到了结果，因为每次往范围内新增数字的时候，唯一可能会得到结果的，肯定是要在新加的这个数字的值的上下范围内，所以使用treeSet带的floor函数和ceiling函数

个人理解，欢迎指导