SlideWindow专栏：leetcode#220 存在重复元素Ⅲ

Question:

给你一个整数数组 nums 和两个整数 k 和 t 。请你判断是否存在 两个不同下标 i 和 j，使得 abs(nums[i] - nums[j]) <= t ，同时又满足 abs(i - j) <= k 。

如果存在则返回 true，不存在返回 false。

Example:

输入：nums = [1,2,3,1], k = 3, t = 0
输出：true

Resolution:

对于序列中的每个元素x，我们都一律只考虑元素x的左侧至多k个元素，而不考虑右侧的元素。我们使用一个有序集合TreeSet存储元素x的左侧至多k个元素，每次移动元素x的时候，维护该有序集合，使其只存储当前元素x的左侧至多k个元素，每次移动后，在该有序集合中查找有没有元素位于[x-t, x+t]范围内，如果有则找到符合题意的答案，返回true，没有则继续移动x，维护最大为k的滑动窗口，直到移动到序列尾。

对于上述思路，我们有两个关键问题：

1. 为什么用有序集合TreeSet去维护[0,k]的这么一个滑动窗口：

（1）我们需要找到在[x-t,x+t]范围内的元素，如果直接查找的话，时间复杂度在o(n*min(n,k)),所以需要对滑动窗口内的元素进行排序，将时间复杂度降低至o(n*log(min(n,k)))；

  (2)  该滑动窗口需要随时增减元素，且元素至多是k，有可能小于k，且是逐渐增加起来的，我们排除掉使用数组的可能

综上所述，我们选择了TreeSet这个数据结构，可随时增减元素，且增加，删除，查找的时间复杂度均为o(logmin(n,k))。

2. 我们如何快速判断滑动窗口内是否存在一个元素位于[x-t,x+t]中：

（1）如果滑动窗口中存在元素等于x，那么直接返回true 

（2）我们只需要找到大于等于x-t的最小元素，并判断它是否小于等于x+t，两个条件都满足的情况下，存在符合题意的答案；

为了防止整型溢出，我们使用long型来存储变量。

Code:

 

    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
        int n = nums.length;
        TreeSet<Long> set = new TreeSet<>();
        for (int i = 0; i < nums.length; i++) {
            Long temp = set.ceiling((long)nums[i] - (long)t);
            if (set.contains((long)nums[i]) || (temp != null && temp <= (long)nums[i] + (long)t))
                return true;
            set.add((long)nums[i]);
            if (set.size() > k)
                set.remove((long)nums[i - k]);
        }
        return false;
    }