题目
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007
解题思路
使用归并排序的想法,把数组分成2个,归并时直接比对2个数组之间,即可得出多少逆序对。随即把2个数组归并排序,合并之后的比较。
代码如下
import java.util.*;
public class Solution {
public int InversePairs(int[] array) {
int[] cnt = {0};
InversePairsHelper(array, 0, array.length - 1, cnt);
return cnt[0];
}
//辅助函数,归并数组
public void InversePairsHelper(int[] array, int left, int right, int[] cnt) {
if (left < right) {
int mid = (left + right) >> 1;
InversePairsHelper(array, left, mid, cnt);
InversePairsHelper(array, mid + 1, right, cnt);
InversePairsMeger(array, left, mid, right, cnt);
}
}
public void InversePairsMeger(int[] array, int left, int mid, int right, int[] cnt) {
//把原数组分为左右2个数组
int[] L = Arrays.copyOfRange(array, left, mid + 1);
int[] R = Arrays.copyOfRange(array, mid + 1, right + 1);
int r = 0, l = 0;
//找出2个数组之间其中有多少逆序对
while (r < R.length) {
while (l < L.length) {
if (L[l] > R[r]) {
break;
}
l++;
}
r++;
cnt[0] %= 1000000007;
cnt[0] += L.length - l;
}
//归并排序2个数组
l = 0;
r = 0;
while (l < L.length && r < R.length) {
array[left++] = L[l] > R[r] ? R[r++] : L[l++];
}
if (r == R.length) {
while (left <= right) {
array[left++] = L[l++];
}
} else {
while (left <= right) {
array[left++] = R[r++];
}
}
}
}