220. Contains Duplicate III

王偉廷發(fā)布于2019-08-16 10:36 / 1323人閱讀

摘要：如果不是，則在相鄰的兩個(gè)內(nèi)再找。如果相鄰的內(nèi)元素絕對值只差在以內(nèi)，說明我們知道到了，返回為了保證，我們在時(shí)，刪除對應(yīng)的的元素都會(huì)落在里。為了解決這個(gè)問題，所有元素橫移。

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.

題意找到一組數(shù)，nums[i] and nums[j] 他們idx差值在k以內(nèi)， 即 j - i <= k.
他們的差的絕對值在t以內(nèi)，即 Math.abs(nums[i] - nums[j]) <= t.
我們可以構(gòu)建一個(gè)大小為t+1的bucket, 比如[0, 1, 2, 3, ... , t] 最大絕對值差的兩個(gè)數(shù)就是t和0. 
如果兩個(gè)數(shù)字出現(xiàn)在同一個(gè)Bucket內(nèi)，說明我們已經(jīng)找到了。 如果不是，則在相鄰的兩個(gè)bucket內(nèi)再找。
如果相鄰的bucket內(nèi)元素絕對值只差在t以內(nèi)，說明我們知道到了，返回true.
為了保證j - i <= k，我們在i>=k時(shí)，刪除 nums[i-k]對應(yīng)的Bucket.

public class Solution {
    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
        if( k < 1 || t < 0) return false;
        Map map = new HashMap<>();
        for(int i = 0; i < nums.length; i++){
            // [-t, 0] [0, t] 的元素都會(huì)落在bucket[0]里。
            // 為了解決這個(gè)問題，所有元素橫移Integer.MIN_VALUE。
            long remappedNum = (long) nums[i] - Integer.MIN_VALUE;
            long bucket = remappedNum / ((long)t + 1);
            if(map.containsKey(bucket) 
                ||(map.containsKey(bucket-1) && remappedNum - map.get(bucket-1) <= t)
                    || (map.containsKey(bucket+1) && map.get(bucket+1) - remappedNum <= t) )
                    return true;
            if(i >= k) {
                long lastBucket = ((long) nums[i-k] - Integer.MIN_VALUE) / ((long)t + 1);
                map.remove(lastBucket);
            }
            map.put(bucket,remappedNum);
        }
        
        return false;
    }
}

TreeSet也可以解決這個(gè)問題， 我們首先需要把i-j <=k 的所有元素裝起來， 
然后集合內(nèi)的元素可以保證有序，在里面找最接近nums[i] +/- t的元素就行了。
[11,12,14,15]
tree.floor(13) = 12
tree.floor(12) = 12

tree.ceiling(13) = 14
tree.ceiling(14) = 14

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