Keep Learning

永远年轻,永远热泪盈眶

LeetCode 220. 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 absolute difference between nums[i] and nums[j] is at most t and the absolute difference between i and j is at most k.
  • 地址

问题分析

  • 该题是LeetCode 219. Contains Duplicate II 进阶题目。又多了一个限制条件,不仅要求索引绝对值之差不超过 K, 还要求两数字绝对值之差不超过 t 。
  • 利用 滑动窗口 + TreeSet(底层实现是 BST) 来做,滑动窗口同样是保持窗口内的个数不超过 K个。而 TreeSet 用作存储窗口中的元素,可以直接借助 ceiling()函数floor()函数来定位到相应元素。具体见实现
  • 复杂度: O(NlogK)

经验教训

  • TreeSet 的 使用
    • ceiling(val)函数 返回集合中 >= val 的最小元素
    • ceiling(val)函数 返回集合中 <= val 的最大元素
  • 此题中的数据溢出问题
    • (long) nums[i] + t(long)nums[i] + t 的区别

代码实现

class Solution {
    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
        if (nums == null || nums.length == 0 || k < 0 || t < 0) {
            return false;
        }
        TreeSet<Long> set = new TreeSet<>();
        int left = 0;
        int right = 0;
        while (right < nums.length) {
            if (right - left > k) { //nums[right]的入窗导致窗口大小超过 k 个
                //左侧出窗
                set.remove((long)nums[left]);
                left++;
            }
            //待入窗元素nums[right],检查原窗口中是否存在[nums[right] - t, nums[right] + t]范围内的数
            Long ceilValue = set.ceiling(((long)nums[right] - t));
            if (ceilValue != null && ceilValue <= (long)nums[right] + t) {
                //若有,返回true
                return true;
            }
            //当前元素入窗
            set.add((long)nums[right]);
            //继续扩窗
            right++;
        }
        return false;

    }
}
阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/zjxxyz123/article/details/79962765
个人分类: LeetCode
上一篇LeetCode 219. Contains Duplicate II
下一篇LeetCode 206. Reverse Linked List
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭