220-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.

【分析】

  1. 一直在想用HashMap解决问题,没办法。用了TreeSet(实现了SortedSet接口),SortedSet有个subSet(from, to)方法比较有效

  2. 主要需要比较坐标和元素值,使Set里最多只有k个元素,每加入一个元素,判断是否有两个元素的元素值之差小于t

【算法实现】

public boolean solution(int[] nums, int k, int t) {
    boolean b = false;
    TreeSet<Long> s = new TreeSet<Long>();
    for(int i=0; i<nums.length; i++) {
        TreeSet<Long> sub  = (TreeSet)s.subSet(nums[i]-t, true, nums[i]+t, true);
        if(!sub.isEmpty()) 
            return true;
        if(i >= k) {
            s.remove((long)nums[i-k]);
        }
        s.add((long)nums[i]);
    }  
    return false;
}

 

posted @ 2015-06-03 20:02  hwu_harry  阅读(100)  评论(0编辑  收藏  举报