Java實現 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

class Solution {
    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
   // 滑動窗口結合查找表，此時滑動窗口即爲查找表本身（控制查找表的大小即可控制窗口大小）
    TreeSet<Long> set = new TreeSet<>();
    for (int i = 0; i < nums.length; i++) {
        // 邊添加邊查找
        // 查找表中是否有大於等於 nums[i] - t 且小於等於 nums[i] + t 的值
        Long ceiling = set.ceiling((long) nums[i] - (long) t);
        if (ceiling != null && ceiling <= ((long) nums[i] + (long) t)) {
            return true;
        }
        // 添加後，控制查找表（窗口）大小，移除窗口最左邊元素
        set.add((long) nums[i]);
        if (set.size() == k + 1) {
            set.remove((long) nums[i - k]);
        }
    }
    return false;
    }
}