LeetCode Exercise 13: Contains Duplicate III

来源:互联网 发布:php统计网站访问人数 编辑:程序博客网 时间:2024/06/12 00:20

题名:Contains Duplicate III

难度:Medium

题目描述:


题目思路:

这道题目乍一看很简单,可实际上涉及了时间复杂度和数据溢出问题。一开始尝试C++用两个简单的循环:在遍历整个vector的情况下,把每个元素与之其后的k个元素检测是否满足条件,结果显示超时。

Java中的TreeSet结构很适合这道题,我们可以将一个TreeSet控制成一个大小为k的窗口window,利用TreeSet的ceiling和floor方法(ceiling:返回此 set 中大于等于给定元素的最小元素;如果不存在这样的元素,则返回 null;floor:返回此 set 中小于等于给定元素的最大元素;如果不存在这样的元素,则返回 null),可避免在序号范围为k内的循环检测。

根据数学知识我们可以把绝对值不等式 | nums[i] - nums[j] | <= t 化简成 nums[i] - t <= nums[j] <= nums[i] + t,对每个vector元素nums[i],我们只需要在window中查看判断是否有元素满足nums[j]即可。

需要注意的是 nums[i] - t <= nums[j] <= nums[i] + t 这个式子的上下界溢出问题。
eg:-2147483647 - 3 = 2147483646;   2147483647 + 3 = -2147483646。
这个我没想到好的解决办法,只是加了两个判断条件。


代码如下:

class Solution {    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {              if (nums == null || k <= 0) {            return false;        }        TreeSet<Integer> window = new TreeSet<>();        for (int i = 0; i < nums.length; i++) {            Integer floor = window.floor(nums[i] + t);            Integer ceil = window.ceiling(nums[i] - t);                        if(nums[i] < 0 && nums[i] - t > 0){  //下界溢出,只考虑上界                if(floor != null){                    return true;                }            }            else if(nums[i] > 0 && nums[i] + t < 0){ //上界溢出,只考虑下界                if(ceil != null){                    return true;                }            }            else if (floor != null && ceil != null && ceil <= floor){ //无溢出                return true;            }            window.add(nums[i]);            if (i >= k) {                window.remove(nums[i - k]);            }        }        return false;    }        }


原创粉丝点击