Contains Duplicate III —— LeetCode

Given an array of integers, find out whether there are two distinct indices i and j in the array such that the difference between nums[i] and nums[j] is at most t and the difference between i and j is at most k.

题目大意：给定一个数组，找出是否存在两个不同下标的值相差<=t,下标i和j相差<=k。

解题思路：题目没说数组有序，那就得按照无序处理，可以排序，然后从头遍历；或者用个BST之类的有序数据结构，遍历数组的时候重建一下，重建的过程中判断是否有合法的解，有就返回true，否则重建完之后返回false。

public class Solution {
    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
        if(nums==null||nums.length==0||k<=0){
            return false;
        }
        TreeSet<Integer> tree = new TreeSet<>();
        for(int i=0;i<nums.length;i++){
            Integer big = tree.floor(nums[i]+t);//floor返回集合里<=指定元素的最大值
            Integer small = tree.ceiling(nums[i]-t);//celing返回集合里>=指定元素的最小值
            if((big!=null&&big>=nums[i])||(small!=null&&small<=nums[i])){
                return true;
            }
            if(i>=k){
                tree.remove(nums[i-k]);
            }
            tree.add(nums[i]);
        }
        return false;
    }
}