原题
给你一个整数数组 nums 和两个整数 k 和 t 。请你判断是否存在 两个不同下标 i 和 j,使得 abs(nums[i] - nums[j]) <= t ,同时又满足 abs(i - j) <= k 。
如果存在则返回 true,不存在返回 false。
Given an integer array nums and two integers k and t, return true if there are two distinct indices i and j in the array such that abs(nums[i] - nums[j]) <= t and abs(i - j) <= k.
Example 1:
Input: nums = [1,2,3,1], k = 3, t = 0
Output: true
Example 2:
Input: nums = [1,0,1,1], k = 1, t = 2
Output: true
Example 3:
Input: nums = [1,5,9,1,5,9], k = 2, t = 3
Output: false
Constraints:
0 <= nums.length <= 2 * 104
-231 <= nums[i] <= 231 - 1
0 <= k <= 104
0 <= t <= 231 - 1
答案
class Solution {
public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
int n = nums.length;
TreeSet<Long> ts = new TreeSet<>();
for (int i = 0; i < n; i++) {
Long u = nums[i] * 1L;
// 从 树中拿出小于等于u的最接近u的数
Long l = ts.floor(u);
// 从 树中拿出大于等于 u 的最接近 u 的数
Long r = ts.ceiling(u);
if(l != null && u - l <= t) return true;
if(r != null && r - u <= t) return true;
// 将当前数加到树中排序,并移除下标范围不在 [max(0, i - k), i) 的数(维持滑动窗口大小为 k)
ts.add(u);
if (i >= k)
ts.remove(nums[i - k] * 1L);
}
return false;
}
}
解析
- treeSet数据结构是红黑树,也即自平衡的排序二叉树,数据按照大小排序在树中,插入数据的时间复杂度logn,查找的时间复杂度logn,所以该答案的算法的时间复杂度为nlogn
- 遍历所有数据,每次遍历都将数据存放到树中,存放的同时将超出范围的值从树中去除
- 每次遍历时还要判断当前范围内是否已经有有效值出现了
- 利用红黑树的的特性,存放到红黑树中的数据已经做好了排序了,所以可以快速的取出红黑树中在某个范围内的值
- 取出距离当前值最近的两个数(因为新加的数才可能会出现有效值,先前的数字已经全部计算过了),看是否符合条件,符合条件即可得到结果
思考
- 抽象化该问题,其实就是要判断数组中,是否存在这么一个范围(k),在这个范围里面,有那么两个数字,他们的差值非常靠近(t)
- 所以问题很简单,首先就是要找出所有符合k的范围,其实就是一个遍历放到tree中,tree中的个数超出(k)后就移除最前面的那一个
- 接着就是找出范围内的有效值了,如果一个个匹配的话那么速度未免也太慢了,使用排序算法排完了序后再去找目标值就快很多了,所以这里就用了treeSet的结构,利用红黑树可以快速的完成排序以及查找。
- 查找目标值的时候这里也很巧妙,只查找了一次就得到了结果,因为每次往范围内新增数字的时候,唯一可能会得到结果的,肯定是要在新加的这个数字的值的上下范围内,所以使用treeSet带的floor函数和ceiling函数
个人理解,欢迎指导