剑指offer 36题 【时间空间效率的平衡】数组中的逆序对

题目描述

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

牛客传送门：点击打开链接

输入描述:

题目保证输入的数组中没有的相同的数字
数据范围：
	对于%50的数据,size<=10^4
	对于%75的数据,size<=10^5
	对于%100的数据,size<=2*10^5

输入例子:

1，2，3，4，5，6，7，0

输出例子:

7

思路：

Ⅰ.由于数组大小最大为200000，假设数组是倒序的，那么逆序对为(1+200000-1)*(200000-1)/2个，约为2*10^10，大于int的范围，所以用long来存储结果。

Ⅱ.在归并排序算法基础上计算逆序对数。例如合并A段和B段时（A、B段都是有序的），创建俩个指针i、j，它们都指向A、B段的最前面，若array[i] > array[j],则说明A段剩余数字都比array[j]大,故count+=A段剩余数字，然后将最小数字array[j]入辅助数组，j++。若array[i] <= array[j]，则array[i]都不大于B段剩余数字，那么count不变，i++。

例如：

①：A段：【2，3，5，6】，B段：【2，4，7，9】，array[i] <= array[j]，A段的2与B段的2，4，7，9都不构成逆序对，那么count不变。同时将最小值A段的2入辅助数组COPY：【2】。

②：接下来A：【3，5，6】，B：【2，4，7，9】，此时A段的3大于B段的4，则A段的3，5，6都与B段的2构成，count值+=A段剩余长度，即count+=3。辅助数组COPY：【2，2】

③：接下来A：【3，5，6】，B：【4，7，9】，此时情况与①一致，count不变。COPY：【2，2，3】

④：接下来A：【5，6】，B：【4，7，9】，此时情况同②一致，count+=A段的剩余长度，即count+=2。COPY：【2，2，3，4】

⑤：接下来A：【5，6】，B：【7，9】，此时情况同①一致，count不变。COPY：【2，2，3，4，5】

⑥：接下来A：【6】，B：【7，9】，此时情况同①一致，count不变。COPY：【2，2，3，4，5，6】

⑦：此时A段结束，将B段全部入辅助数组。

代码如下：

public class Title36 {
	long count; // 逆序对数
	int[] copy; // 辅助数组
	public int InversePairs(int [] array) {
        if(array == null || array.length == 0)
        	return 0;
        // 初始化
        count = 0;
        copy = new int[array.length];
        mergeSort(array,0,array.length -1);
        return (int)(count%1000000007);
    }

	public void mergeSort(int[] array,int begin,int end){
		if(begin < end){
			int mid = (begin + end) >>> 1;
			mergeSort(array,begin,mid);
			mergeSort(array,mid+1,end);
			merge(array,begin,mid,end);
		}
	}
	
	public void merge(int[] array,int begin,int mid,int end){
		int i=begin,j=mid+1,index = begin;
		for(;i<=mid && j<=end;){
			if(array[i] > array[j]){
				count += mid-i+1;
				copy[index++] = array[j++];
			}else{
				copy[index++] = array[i++];
			}
		}
		
		while(i <= mid)
			copy[index++] = array[i++];
		while(j <= end)
			copy[index++] = array[j++];
		
		for(int k=begin;k<=end;k++)
			array[k] = copy[k];
	}
	
	public static void main(String[] args) {
		int[] array = {1,2,3,4,5,6,7,0};
		Title36 clazz = new Title36();
		System.out.println(clazz.InversePairs(array)); // 7
		
		
		int[] arrayTwo = {2,3,5,6,1,4,5,7};
		System.out.println(clazz.InversePairs(arrayTwo));// 7
	}
}