220 Contains Duplicate III

题目链接：https://leetcode.com/problems/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.

解题思路：

滑动窗口 + TreeSet

这个题自己没有想出来。。。大概能想到数据结构是 Tree，再加上滑动窗口的概念。但是没有很好地把它们整合到一起。

参考LeetCode Discuss：https://leetcode.com/discuss/38177/java-o-n-lg-k-solution TreeSet 数据结构（Java）使用红黑树实现，是平衡二叉树的一种。 该数据结构支持如下操作：

1. floor() 方法返 set 中 ≤ 给定元素的最大元素；如果不存在这样的元素，则返回 null。

2. ceiling() 方法返回 set 中 ≥ 给定元素的最小元素；如果不存在这样的元素，则返回 null。

参考链接：http://bookshadow.com/weblog/2015/06/03/leetcode-contains-duplicate-iii/

public class Solution {
public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
if(k < 1 || t < 0)
return false;
TreeSet<Integer> set = new TreeSet<Integer>();
for(int i = 0; i < nums.length; i++){
int n = nums[i];
if(set.floor(n) != null && n <= t + set.floor(n) ||
set.ceiling(n) != null && set.ceiling(n) <= t + n)
return true;
set.add(n);
if (i >= k)
set.remove(nums[i - k]);
}
return false;
}
}

29 / 29 test cases passed.
Status: Accepted
Runtime: 504 ms

还有另一种解法，思路差不多，但更加巧妙的运用了 TreeSet 的方法。

SortedSet subSet(E fromElement, E toElement)

返回此 set 的部分视图，其元素从 fromElement（包括）到 toElement（不包括）。

参考链接：http://www.programcreek.com/2014/06/leetcode-contains-duplicate-iii-java/

import java.util.SortedSet;
public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
if (k < 1 || t < 0)
return false;
SortedSet<Long> set = new TreeSet<Long>();
for (int j = 0; j < nums.length; j++) {
long leftBoundary = (long) nums[j] - t;
long rightBoundary = (long) nums[j] + t + 1;
SortedSet<Long> subSet = set.subSet(leftBoundary, rightBoundary);
if (!subSet.isEmpty())
return true;
set.add((long) nums[j]);
if (j >= k) {
set.remove((long) nums[j - k]);
}
}
return false;
}

29 / 29 test cases passed.
Status: Accepted
Runtime: 524 ms

注意：

一开始提交代码时，显示编译错误 cannot find symbol: class SortedSet。上网找了找原因，发现是需要自己导入包，leetcode 上本身不具有。因此多加了一句 import java.util.SortedSet;