#220. 存在重复元素 III
题目描述
给定一个整数数组,判断数组中是否有两个不同的索引 i 和 j,使得 nums [i] 和 nums [j] 的差的绝对值最大为 t,并且 i 和 j 之间的差的绝对值最大为 ķ。
示例 1:
输入: nums = [1,2,3,1], k = 3, t = 0
输出: true
示例 2:
输入: nums = [1,0,1,1], k = 1, t = 2
输出: true
示例 3:
输入: nums = [1,5,9,1,5,9], k = 2, t = 3
输出: false
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/contains-duplicate-iii
分析
根据长度限制建立滑动窗口,滑动窗口内的数字用红黑树(Java的treeset就是红黑树实现的)存贮,利用Java的天花板和地板函数进行二分查找(成本低)。因为红黑树动态增减数字时,维护有序的成本低。
ceiling(key) 函数:返回大于等于 key 的最小元素,如果不存在,返回空。下面的是这个函数的文档(通过 Intellij IDEA 查看)。
floor(key) 函数:返回小于等于 key 的最大元素,如果不存在,返回空。下面的是这个函数的文档(通过 Intellij IDEA 查看)。
代码
//排序
public static boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
if (t < 0 || k == 10000){
return false;
}
TreeSet<Long> set = new TreeSet<>();
//建立滑动窗口
for (int i = 0; i < nums.length; i++) {
Long floor = set.floor(((long)nums[i] + t));
Long ceiling = set.ceiling(((long)nums[i] - t));
if (floor != null && floor >= nums[i]
|| ceiling != null && ceiling <= nums[i]){
return true;
}
set.add((long)nums[i]);
if (set.size() > k){
set.remove((long)nums[i-(k)]);
}
}
return false;
}