Leetcode ---220. 存在重复元素 III （数组）

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

JAVA

思路：

1. 平衡二叉树
2. 初始化一颗空的二叉搜索树 set
3. 对于每个元素xxx，遍历整个数组
   在 set 上查找大于等于xxx的最小的数，如果s−x≤t则返回 true
   在 set 上查找小于等于xxx的最大的数，如果x−g≤t则返回 true
   在 set 中插入x
   如果树的大小超过了k, 则移除最早加入树的那个数。
4. 返回 false

• floor(E e) 方法返回在这个集合中小于或者等于给定元素的最大元素，如果不存在这样的元素,返回null.

• ceiling(E e) 方法返回在这个集合中大于或者等于给定元素的最小元素，如果不存在这样的元素,返回null.

• 参数：e这是要匹配的值。

class Solution {
    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
        TreeSet<Integer> set = new TreeSet<>();//学会只用 treeset
        for (int i = 0; i < nums.length; ++i) {
            // Find the successor of current element
            Integer s = set.ceiling(nums[i]); //刚开始因为set中没有元素，为空
            if (s != null && s <= nums[i] + t) return true;

            // Find the predecessor of current element
            Integer g = set.floor(nums[i]);
            if (g != null && nums[i] <= g + t) return true;

            set.add(nums[i]);
            if (set.size() > k) { //超过k的时候，移走前一个元素
                set.remove(nums[i - k]);
            }
        }
        return false;
    }
}