标签：
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.
区别于上一题，这道不仅有index距离的限制，还加上了两值和的要求。这就要求选择的数据结构能够支持快速的按值搜索。选择TreeSet是因为它内部是用红黑树实现的，所有节点按照natural order从低到高进行排序，可以提供O(logn)的add/remove/contain的操作，并且有floor和ceiling两个method，可用于在一定range进行快速查找。Leetcode上有一个test case测的是Integer值溢出变复数，所以在下面的第二种解法里，用了Long存值，而且要注意int到long类型的cast。
另外一点关于TreeSet需要注意的是，它基于Comparable这个interface进行实现，而Set是基于equals。所以需要考虑compareTo的实现，以保证两者的统一。相关的文档里也作出了强调。
解法一：
1 public static boolean containsNearbyAlmostDuplicate1(int[] nums, int k, int t){ 2 if (k<1 || t<0) return false; 3 4 TreeSet<Integer> set = new TreeSet<Integer>(); 5 for (int i=0;i<nums.length;i++){ 6 int current = nums[i]; 7 if((set.floor(current)!=null && current<= set.floor(current)+t)|| 8 (set.ceiling(current)!=null)&& current>=set.ceiling(current)-t ) 9 return true; 10 set.add(current); 11 12 if(i>=k) 13 set.remove(nums[i-k]); 14 } 15 return false; 16 } 17
解法二：
1 public static boolean containsNearbyAlmostDuplicate2(int[] nums, int k, int t){ 2 if (k<1 || t<0) return false; 3 4 TreeSet<Long> set = new TreeSet<Long>(); 5 for(int i=0;i<nums.length;i++){ 6 int current = nums[i]; 7 SortedSet<Long> subset = set.subSet((long)current-t, (long)current+t+1); 8 9 if(!subset.isEmpty()) return true; 10 set.add((long) current); 11 if(i >=k) set.remove((long)nums[i-k]);//remember to cast it to long 12 } 13 return false; 14 }
Reference：http://www.programcreek.com/2014/06/leetcode-contains-duplicate-iii-java/
LeetCode – Remove Duplicates from Sorted List III (Java)
标签：
原文地址：http://www.cnblogs.com/7070roro/p/4666095.html
