#leetcode#220. Contains Duplicate III

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/ChiBaoNeLiuLiuNi/article/details/72769778

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.



--------------------------------------


两种解法, 1, bucket。 2, Treeset

相通之处在于都是维护一个长度为K的sliding window来确保 absolute difference between i and j is at most k.

转换成long避免溢出, 比如   Integer.MIN_VALUE (-2147483648, -2147483647)k = 3, t = 3 这种input


Bucket:O(n)

扫描二维码关注公众号,回复: 4663904 查看本文章
public class Solution {
    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
        if(k < 1 || t < 0)  return false;
        long valueDiff = (long)t;
        long indexDiff = (long)k;
        Map<Long, Long> map = new HashMap<>();
        for(int i = 0; i < nums.length; i++){
            long value = (long)nums[i] - Integer.MIN_VALUE;
            long bucket = value / (valueDiff + 1);
            if(map.containsKey(bucket) || map.containsKey(bucket - 1) && value - map.get(bucket - 1) <= valueDiff
               || map.containsKey(bucket + 1) && map.get(bucket + 1) - value <= valueDiff)
                return true;
            map.put(bucket, value);
            if(i >= k){
                long idx = ((long)nums[i - k] - Integer.MIN_VALUE) / (valueDiff + 1);
                map.remove(idx);
            }
        }
        
        return false;
    }
}

TreeSet: O(nlogk)

public class Solution {
    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {

        TreeSet<Long> set = new TreeSet<>();
        for(int i = 0; i < nums.length; i++){
            Long ceil = set.ceiling((long)nums[i] - (long)t);
            Long floor = set.floor((long)nums[i] + (long)t);
            if(ceil != null && ceil <= nums[i] || floor != null && floor >= nums[i])
                return true;
            set.add((long)nums[i]);
            if(i >= k)
                set.remove((long)nums[i - k]);
        }
        
        return false;
    }
}


猜你喜欢

转载自blog.csdn.net/ChiBaoNeLiuLiuNi/article/details/72769778