题目
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007
思路
归并排序求数组中逆序对个数。复习一下归并排序的代码。
代码
public class Solution {
long cnt=0;
public void mergesort(int[] arr, int L, int R) {
if(L == R) {
return;
}
int mid = L + ((R - L) >> 1);
mergesort(arr, L, mid);
mergesort(arr, mid + 1, R);
merge(arr, L, mid, R);
}
public void merge(int[] arr, int L, int mid, int R){
int[] tmp = new int[R-L+1];
int k=0;
int i=L;
int j=mid+1;
while(i<=mid && j<=R){
if(arr[i]>arr[j]){
cnt+=mid-i+1;
if(cnt>1000000007)
cnt%=1000000007;
tmp[k++]=arr[j++];
}
else
tmp[k++]=arr[i++];
}
while(i<=mid)
tmp[k++]=arr[i++];
while(j<=R)
tmp[k++]=arr[j++];
for(k=0;k<tmp.length;k++)
arr[L+k]=tmp[k];
}
public int InversePairs(int [] array) {
mergesort(array, 0, array.length-1);
return (int)cnt;
}
}