存在重复元素 III(Java算法每日一题)

问:给你一个整数数组 nums 和两个整数 k 和 t 。请你判断是否存在 两个不同下标 i 和 j,使得 abs(nums[i] - nums[j]) <= t ,同时又满足 abs(i - j) <= k 。
如果存在则返回 true,不存在返回 false。

例:
输入:nums = [1,2,3,1], k = 3, t = 0
输出:true

答:
暴力算法

class Solution {
    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
        int len = nums.length;
        for(int i = 0;i < len;i++)
            for(int j = i+1;j < len;j++)
            {
                if(Math.abs(((long)nums[i]-(long)nums[j])) <= t && Math.abs(i - j) <= k)
                    return true;
            }
        return false;
    }
}

滑动窗口

class Solution {
    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
        int len = nums.length;//求数组nums的长度
        TreeSet<Long> tr = new TreeSet<Long>();
        for(int i =0;i < len;i++)
        {
            Long u = nums[i] * 1L;//i下标对应的值u
            Long x = tr.floor(u);//u向下取小于u的最大数
            Long y = tr.ceiling(u);//u向上取大于u的最小数

            if(x!=null && u - x <= t)//u-x<=t
                return true;
            if(y!=null && y - u <= t)//y-u<=t
                return true;
            tr.add(u);
            if(i >= k)//如果超过了k,就将i-k的给去掉
                tr.remove(nums[i-k] *1L);
        }
        return false;
    }
}
上一篇:5706. 句子相似性 III


下一篇:剑指 Offer 32 - III. 从上到下打印二叉树 III