493. Reverse Pairs

acrazing 發(fā)布于2019-08-14 17:36 / 1208人閱讀

摘要：題目鏈接和還有是一類題，解法都差不多。可以做，但是這道題如果輸入是有序的，簡(jiǎn)單的會(huì)超時(shí)，所以得用來做。算的方法是比如給的例子，現(xiàn)在分成了左右兩部分，拿兩個(gè)指針和。

493. Reverse Pairs

題目鏈接：
https://leetcode.com/problems...

和Count of Smaller Numbers After Self還有count of range sum是一類題，解法都差不多。BST可以做，但是這道題如果輸入是有序的，簡(jiǎn)單的bst會(huì)超時(shí)，所以得用AVL來做。
然后就是binary index tree的做法，計(jì)算大于nums[j]2的時(shí)候就是拿全部的sum減去sum(nums[j] 2)

public class Solution {
    public int reversePairs(int[] nums) {
        int res = 0;
        int n = nums.length;
        if(n == 0) return res;
        // reflection
        Map map = new HashMap();
        long[] sorted = new long[2*n];
        for(int i = 0; i < n; i++) {
            sorted[2*i] = nums[i];
            sorted[2*i + 1] = (long) nums[i] * 2;
        }
        Arrays.sort(sorted);
        int idx = 1;
        for(long num : sorted) {
            if(!map.containsKey(num)) map.put(num, idx++);
        }
        
        BIT t = new BIT(idx);
        int sum = 0;
        for(int j = 0; j < n; j++) {
            // find how many number > 2 * nums[j]
            long num = (long) nums[j];
            res += sum - t.sum(map.get(num*2));
            t.add(map.get(num), 1);
            sum++;
        }
        return res;
    }
    
    class BIT {
        int n;
        int[] tree;
        BIT(int n) { this.n = n; tree = new int[n]; }
        
        protected int sum(int i) {
            int res = 0;
            while(i > 0) {
                res += tree[i];
                i -= i & -i;
            }
            return res;
        }
        
        protected void add(int i, int val) {
            while(i < n) {
                tree[i] += val;
                i += i & -i;
            }
        }
    }
}

merge sort的做法，同樣是每次要統(tǒng)計(jì)左邊有多少結(jié)果是 > 2 * nums[j]的，每次sort完之后先算有多少pairs再merge。算pairs的方法是：
比如給的例子，現(xiàn)在分成了左右兩部分[1, 1, 2], [3, 3]，拿兩個(gè)指針i和j。

if nums[i]/2.0 > nums[j]，表示所有小與等于nums[j]的值都滿足這個(gè)條件，一直增大到不滿足條件的，最后j - (mid+1)就是全部滿足該條件且包含nums[i]的pair數(shù)目

else, 表示nums[i]小了，需要i++

loop的invariant用包含nums[j]的也行：

ifnums[i]/2.0 > nums[j]表示所有大于等于nums[i]的都滿足條件，res += mid - i + 1，同時(shí)j++

else表示i小了，所以i++

每次重新開一個(gè)aux就超時(shí)了，所以把a(bǔ)ux當(dāng)全局變量，開一次

public class Solution {
    public int reversePairs(int[] nums) {
        int n = nums.length;
        if(n == 0) return 0;
        
        // merge sort
        res = 0;
        aux = new int[n];
        sort(nums, 0, n - 1);
        return res;
    }
    int res;
    int[] aux;
    
    private void sort(int[] nums, int l, int r) {
        if(l >= r) return;
        int mid = l + (r - l) / 2;
        sort(nums, l, mid);
        sort(nums, mid + 1, r);
        int i = l, j = mid + 1;
        while(j <= r) {
            while(i <= mid && nums[i]/2.0 <= nums[j]) i++;
            // count number of pairs include nums[j]
            res += mid - i + 1;
            j++;
        }
        merge(nums, l, mid, r);
    }
    
    private void merge(int[] nums, int l, int mid, int r) {
        for(int k = l; k <= r; k++) aux[k] = nums[k];
        
        int i = l, j = mid + 1;
        for(int k = l; k <= r; k++) {
            if(i > mid) nums[k] = aux[j++];
            else if(j > r) nums[k] = aux[i++];
            else if(aux[i] > aux[j]) nums[k] = aux[j++];
            else nums[k] = aux[i++];
        }
    }
}