《剑指 offer》 学习35之数组中的逆序对

题目描述

在数组中的两个数字，如果前面一个数字大于后面的数字，则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数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];
	    } 
	}
}

测试结果