【Kevin Learn 算法与数据结构】--＞《剑指 offer》数组中的逆序对

题目描述

在数组中的两个数字，如果前面一个数字大于后面的数字，则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007
题目链接：​​​牛客网​​

输入描述:
题目保证输入的数组中没有的相同的数字

数据范围：

对于%50的数据,size<=10^4

对于%75的数据,size<=10^5

对于%100的数据,size<=2*10^5

解题思路

public class Main {
    public static long cnt = 0;
    public static int[] temp;   // 在这里声明辅助数组，而不是在 merge() 递归函数中声明
    
    public static void main(String[] args) {
        int[] data = {1,2,3,4,5,6,7,0};
        System.out.println(inversePairs(data));
    }
    
    public static int inversePairs(int[] nums) {
        temp = new int[nums.length];
        mergeSort(nums,0,nums.length - 1);
        return (int)(cnt % 1000000007);
    }
    
    public static void mergeSort(int[] nums,int l,int h) {
        if (h - l < 1) {
            return;
        } 
        
        int m = l + (h - l) / 2;
        mergeSort(nums, l, m);
        mergeSort(nums, m + 1, h);
        merge(nums, l, m, h);
    }
    
    public static 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) {
                temp[k] = nums[j++];
            }else if (j > h) {
                temp[k] = nums[i++];
            } else if (nums[i] < nums[j]) {
                temp[k] = nums[i++];
            } else {
                temp[k] = nums[j++];
                cnt += m - i + 1;     // nums[i] > nums[j]，说明 nums[i...mid] 都大于 nums[j]
            }
            k++;
        }
        
        for (k = l;k <= h;k++) {
            nums[k] = temp[k];
        } 
    }
}

测试结果