在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007
示例1
输入
1,2,3,4,5,6,7,0
输出
7
这道题我第一眼是用直接比较的方法,从前向后依次遍历,找到前面大于后面的逆序对,记录下来。但是时间复杂度高达n^2,很明显不能通过题目要求。
于是我们换个思路,将数组分成一个个单个数字,数字之间两两比较,然后组成顺序数组,再将顺序数组之间比较。假设有顺序数组 A 和 B ,A 在 B 的前面。那么如果有A的第i个数字比B的第j个数字大,则有i后面的数字都比B的第j个数字大,可以组成A.length-i个逆序对。再将A、B组成新的顺序数组,和后面的比较。可以看出来,这就是在归并排序基础上的一次改进。代码如下:
package cn.lyk.swardOffer.InversePairs;
import java.util.Scanner;
public class Solution {
private int num = 0;
public static void main(String[] args){
Scanner scanner = new Scanner(System.in);
int num = scanner.nextInt();
int[] array = new int[num];
for (int i=0;i<num;i++)
array[i] = scanner.nextInt();
Solution solution = new Solution();
System.out.println(solution.InversePairs(array));
}
public int InversePairs(int [] array) {
return solution2(array);
}
/**
* 方法2:采用归并排序,假设有顺序数组 A 和 B ,A 在 B 的前面。
* 那么如果有A的第i个数字比B的第j个数字大,则有i后面的数字都
* 比B的第j个数字大,可以组成A.length-i个逆序对。
*/
private int solution2(int[] array){
mergeSort(array,0,array.length-1);
return num;
}
private void mergeSort(int[] array, int start, int end){
if(start < end){
int middle = (start+end)/2;
mergeSort(array,start,middle);
mergeSort(array,middle+1,end);
merge(array,start,middle,end);
}
}
private void merge(int[] array, int start, int middle, int end){
int[] tmp = new int[end-start+1];
int k = 0;
int i = start;
int j = middle+1;
while (i<=middle && j<=end){
if(array[i]<array[j]){
tmp[k++] = array[i++];
}
else {
num+=middle-i+1;
num = num%1000000007;
tmp[k++] = array[j++];
}
}
while (i<=middle) {
tmp[k++] = array[i++];
}
while (j<=end)
tmp[k++] = array[j++];
for(int m=0;m<tmp.length;m++){
array[start+m] = tmp[m];
}
}
}