《剑指Offer》51. 数组中的逆序对

题目链接

牛客网

题目描述

在数组中的两个数字，如果前面一个数字大于后面的数字，则这两个数字组成一个逆序对。输入一个数组，求出这个数组中的逆序对的总数。

解题思路

就是在归并排序的时候做一点手脚，记录一下cnt数量
如果不用成员变量的tmp数组，每次merge的时候创建，会超时

public class Solution {
    private long cnt = 0; // 注意cnt是int可能越界
    private int[] tmp; // 在这里声明辅助数组，而不是在 merge() 递归函数中声明
    public int InversePairs(int [] nums) {
        if (nums==null || nums.length==0) return 0;
        tmp = new int[nums.length]; // 注意并不是=nums
        mergeSort(nums, 0, nums.length-1);
        return (int)(cnt%1000000007);
    }
    private void mergeSort(int[] nums, int l, int h) {
        if (h<=l) return;
        int m = l+(h-l)/2;
        mergeSort(nums, l, m); // 左侧排序
        mergeSort(nums, m+1, h); // 右侧排序
        merge(nums, l, m, h); // 左右连接起来并由tmp更改nums
    }
    private void merge(int[] nums, int l, int m, int h) {
        int i = l, j = m+1, k = l;
        while (i<=m || j<=h) {
            if (i>m) tmp[k] = nums[j++];
            else if (j>h) tmp[k] = nums[i++];
            else if (nums[i]<=nums[j]) tmp[k] = nums[i++];
            else {
                tmp[k] = nums[j++];
                cnt += m-i+1;
            }
            k++;
        }
        for (k=l;k<=h;k++) { // 更改nums
            nums[k] = tmp[k];
        }
    }
}