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.
Subscribe to see which companies asked this question
函数 subSet
public SortedSet<E> subSet(E fromElement,E toElement)
参数
-
fromElement-- 这是返回集合的低端点(包括)。
-
toElement-- 这是返回集合的高点(不包括)。
</pre><pre code_snippet_id="1708771" snippet_file_name="blog_20160605_6_7047732" name="code" class="java">
</pre><pre code_snippet_id="1708771" snippet_file_name="blog_20160605_6_7047732" name="code" class="java">
</pre><pre code_snippet_id="1708771" snippet_file_name="blog_20160605_6_7047732" name="code" class="java">
//Set的用法 用两个for循环会超时O(nk). 使用treeset( backed by binary search tree) 的subSet函数,可以快速搜索. 复杂度为 O(n logk)
public class Solution {
public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
//input check
if(k<1 || t<0 || nums==null || nums.length<2) return false;
SortedSet<Long> set = new TreeSet<Long>();
for(int j=0; j<nums.length; j++) {
SortedSet<Long> subSet = set.subSet((long)nums[j]-t, (long)nums[j]+t+1);
if(!subSet.isEmpty()) return true;
if(j>=k) {
set.remove((long)nums[j-k]);
}
set.add((long)nums[j]);
}
return false;
}
}
/*
* TreeSet存储对象的时候, 可以排序, 但是需要指定排序的算法
*
* Integer能排序(有默认顺序), String能排序(有默认顺序), 自定义的类存储的时候出现异常(没有顺序)
*
* 如果想把自定义类的对象存入TreeSet进行排序, 那么必须实现Comparable接口
* 在类上implement Comparable
* 重写compareTo()方法
* 在方法内定义比较算法, 根据大小关系, 返回正数负数或零
* 在使用TreeSet存储对象的时候, add()方法内部就会自动调用compareTo()方法进行比较, 根据比较结果使用二叉树形式进行存储
*/