《leetCode》：Contains Duplicate III

题目

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.

思路一：报超时

直接借用两层for循环来解决。

public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
        for(int i=0;i<nums.length;i++){
            for(int j=i+1;j<=i+k&&j<nums.length;j++){
                long r = nums[i]-nums[j];
                if(Math.abs(r)<=t){
                    return true;
                }
            }
        }
        return false;
    }

在上面的代码中如果不借用long类型来存储结果，则会在下面的测试用例中产生溢出。

上面程序的时间复杂度为O(n*k)，其中n为数组的长度。既然不能AC，则需要寻找更好的算法来解决。

思路二

j-i<=k;
思路来自于：http://blog.csdn.net/xudli/article/details/46323247

     public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {  
        if(nums==null||k<=0 ||t<0){
            return false;
        }
        SortedSet<Long> set = new TreeSet<Long> ();
        for(int i=0;i<nums.length;i++){
            SortedSet<Long> s = set.subSet((long)nums[i]-t, (long)nums[i]+t+1);
            if(!s.isEmpty()){
                return true;
            }
            if(k<=i){//始终维持set中的元素个数为k个
                set.remove((long)nums[i-k]);
            }
            set.add((long)nums[i]);

        }
        return false;
    }