(leetcode)contains duplicateIII[java]

题目描述:Given an array of integers, find out whether there are two distinct indices i and j in the array such that the difference between nums[i] and nums[j] is at most t and the difference between i and j is at most k.

这道题一眼看上去就知道要对一个滑动窗口为k的数组进行搜索,然后寻找与当前元素相差<t的元素。如果使用暴力搜索的话,时间复杂度为O(Nk)(大家可以算下)这个在k特别大的时候会TLE,所以必须对滑动窗口进行处理,理想情况下是组织成一棵树,而且最好是一个平衡二叉树。这样对元素的查找、删除和添加都是LOG(K)时间。如果用C实现,比较麻烦,所幸java中有treeset这个类,可以生成一个平衡二叉树而无需考虑实现细节。

代码如下,这里面注意两点:1、leetcode中没有SortedSet,需要自己import,2、在nums[i]+t+1时可能发生溢出,需要将其强制转换为long类型

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

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值