4-8 二分搜尋樹底層實現的順序性 Contain Duplicate III

題目: LeetCode 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

import java.util.TreeSet;

// 220. Contains Duplicate III
// https://leetcode.com/problems/contains-duplicate-iii/description/
// 時間複雜度: O(nlogk)
// 空間複雜度: O(k)
public class Solution {

    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {

        // 這個問題的測試資料在使用int進行加減運算時會溢位
        // 所以使用long long
 

        TreeSet<Long> record = new TreeSet<Long>();
        for(int i = 0 ; i < nums.length ; i ++){

            if(record.ceiling((long)nums[i] - (long)t) != null &&
                    record.ceiling((long)nums[i] - (long)t) <= (long)nums[i] + (long)t)
            return
 
 true;

            record.add((long)nums[i]);

            if(record.size() == k + 1)
                record.remove((long)nums[i-k]);
        }

        return false;
    }

    private static void printBool(boolean b){
        System.out.println(b ? "True" : "False");
    }

    public static void main(String[] args) {

        int[] nums = {-2147483648, -2147483647};
        int k = 3;
        int t = 3;
        printBool((new Solution()).containsNearbyAlmostDuplicate(nums, k, t));
    }
}