[LeetCode] 220. Contains Duplicate III Java

题目:

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

题意及分析:

给出一个数组,要求判断是否存在,对于其中两个不同的元素有:(1) |i-j| <= k (2) |nums[i]-nums[j]| <= t。

TreeSet(binary search tree)  o(Nlogk)

利用treeset,用floor方法查找treeSet中大于等于当前值且与当前元素距离小于t的最大元素,然后如果该元素不为空,那么就可以找到符合题意的两个元素。

或者用用ceiling方法查找treeSet中大于等于当前值且与当前元素距离小于t的最小元素,然后如果该元素不为空,那么就可以找到符合题意的两个元素。

最后需要注意的一点,就是两个元素之间的距离要小于等于k,即|i-j|<=k,所以当前treeSet中最多只能有k个元素。

代码:

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

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

           final Long floor = treeSet.floor((long)nums[i]+t);
           final Long ceil = treeSet.ceiling((long)nums[i]-t);

           if((floor!=null&&floor>=(long)nums[i])||
                   (ceil!=null&&ceil<=(long)nums[i])){
               return true;
           }

           treeSet.add((long)nums[i]);
           if(i>=k){        //因为元素的坐标差不能超过k,所以在treeSet中最多只能有k个元素
               treeSet.remove((long)nums[i-k]);
           }
       }
        return false;
    }
}

   

 

posted @ 2017-09-14 16:54  荒野第一快递员  阅读(723)  评论(0编辑  收藏  举报