剑指offer_057 值和下标之差都在给定范围内

题目：

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

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

示例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

提示：

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) {
        TreeSet<Long> set = new TreeSet<>();
        for (int i = 0; i < nums.length; i++) {
            Long c = set.ceiling((long)nums[i] - (long)t);
            if (c != null && c <= ((long)nums[i] + (long)t))
                return true;
            set.add((long)nums[i]);
            if (i >= k) {
                set.remove((long)nums[i - k]);
            }
        }
        return false;
    }
}

解题思路：

首先应该想到这道题应该是通过滑动窗口去解题，把每个数放到集合中去，并且维持滑动窗口的长度不会超过 k 。然后就是要想办法把 abs(nums[i] - nums[j]) <= t 这个条件满足，比如说当前数是 5 ，t = 2， 只需要另一个数大于等于 3， 小于等于 7 就可以满足这个条件，那么有了 5 以后，另一个数大于等于 3 的数应该怎么确定呢，就可以使用 TreeSet 中的方法 ceiling ，可以返回大于等于当前数值的最小数。这道题就迎刃而解啦。

注意：

好好吃饭。

参考链接：

力扣