LeetCode Exercise 13: Contains Duplicate III

题名：Contains Duplicate III

难度：Medium

题目描述：

题目思路：

这道题目乍一看很简单，可实际上涉及了时间复杂度和数据溢出问题。一开始尝试C++用两个简单的循环：在遍历整个vector的情况下，把每个元素与之其后的k个元素检测是否满足条件，结果显示超时。

Java中的TreeSet结构很适合这道题，我们可以将一个TreeSet控制成一个大小为k的窗口window，利用TreeSet的ceiling和floor方法（ceiling：返回此 set 中大于等于给定元素的最小元素；如果不存在这样的元素，则返回 null；floor：返回此 set 中小于等于给定元素的最大元素；如果不存在这样的元素，则返回 null），可避免在序号范围为k内的循环检测。

根据数学知识我们可以把绝对值不等式 | nums[i] - nums[j] | <= t 化简成 nums[i] - t <= nums[j] <= nums[i] + t，对每个vector元素nums[i]，我们只需要在window中查看判断是否有元素满足nums[j]即可。

需要注意的是 nums[i] - t <= nums[j] <= nums[i] + t 这个式子的上下界溢出问题。

eg：-2147483647 - 3 = 2147483646；   2147483647 + 3 = -2147483646。

这个我没想到好的解决办法，只是加了两个判断条件。

代码如下：

class Solution {
    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {      
        if (nums == null || k <= 0) {
            return false;
        }

        TreeSet<Integer> window = new TreeSet<>();
        for (int i = 0; i < nums.length; i++) {
            Integer floor = window.floor(nums[i] + t);
            Integer ceil = window.ceiling(nums[i] - t);
            
            if(nums[i] < 0 && nums[i] - t > 0){  //下界溢出，只考虑上界
                if(floor != null){
                    return true;
                }
            }
            else if(nums[i] > 0 && nums[i] + t < 0){ //上界溢出，只考虑下界
                if(ceil != null){
                    return true;
                }
            }
            else if (floor != null && ceil != null && ceil <= floor){ //无溢出
                return true;
            }

            window.add(nums[i]);
            if (i >= k) {
                window.remove(nums[i - k]);
            }
        }

        return false;
    }
        
}