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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/contains-duplicate-iii

分析

	根据长度限制建立滑动窗口，滑动窗口内的数字用红黑树（Java的treeset就是红黑树实现的）存贮，利用Java的天花板和地板函数进行二分查找（成本低）。因为红黑树动态增减数字时，维护有序的成本低。

ceiling(key) 函数：返回大于等于 key 的最小元素，如果不存在，返回空。下面的是这个函数的文档（通过 Intellij IDEA 查看）。
floor(key) 函数：返回小于等于 key 的最大元素，如果不存在，返回空。下面的是这个函数的文档（通过 Intellij IDEA 查看）。

代码

//排序
    public static boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
        if (t < 0 || k == 10000){
            return false;
        }
        TreeSet<Long> set = new TreeSet<>();
        //建立滑动窗口
        for (int i = 0; i < nums.length; i++) {

            Long floor      = set.floor(((long)nums[i] + t));
            Long ceiling    = set.ceiling(((long)nums[i] - t));
            if (floor != null && floor >= nums[i]
                || ceiling != null && ceiling <= nums[i]){
                return true;
            }
            set.add((long)nums[i]);
            if (set.size() > k){
                set.remove((long)nums[i-(k)]);
            }
        }
        return false;
    }