题目描述: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.
这道题一眼看上去就知道要对一个滑动窗口为k的数组进行搜索,然后寻找与当前元素相差<t的元素。如果使用暴力搜索的话,时间复杂度为O(Nk)(大家可以算下)这个在k特别大的时候会TLE,所以必须对滑动窗口进行处理,理想情况下是组织成一棵树,而且最好是一个平衡二叉树。这样对元素的查找、删除和添加都是LOG(K)时间。如果用C实现,比较麻烦,所幸java中有treeset这个类,可以生成一个平衡二叉树而无需考虑实现细节。
代码如下,这里面注意两点:1、leetcode中没有SortedSet,需要自己import,2、在nums[i]+t+1时可能发生溢出,需要将其强制转换为long类型
import java.util.SortedSet;
public class Solution {
public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
if(nums.length<2||k<1||t<0)
return false;
SortedSet<Long> binaryTree = new TreeSet<Long>();
for(int i = 0;i<nums.length;i++){
SortedSet<Long> sonSet = binaryTree.subSet((long)nums[i]-t,(long)nums[i]+t+1);
if(!sonSet.isEmpty())
return true;
if(i>=k)
binaryTree.remove((long)nums[i-k]);
binaryTree.add((long)nums[i]);
}
return false;
}
}