LeetCode-220:存在重复元素 [Java实现]

2024年6月10日 50次阅读

给定一个整数数组，判断数组中是否有两个不同的索引 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

这一题用滑动窗口实现的话，复杂度是n*k，提交之后会超时。使用二叉排序树可以有效降低时间复杂度，如下

class Solution {
  public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
        if (k < 1 || t < 0)
            return false;
        SortedSet<Long> binTree = new TreeSet<>();
        for (int i = 0; i < nums.length; i++) {
            SortedSet<Long> son = binTree.subSet((long) nums[i] - t, (long) nums[i] + t + 1);
            if (!son.isEmpty())
                return true;
            if (i >= k)
                binTree.remove((long) nums[i - k]);
            binTree.add((long) nums[i]);
        }
        return false;
    }
}
点赞