题目:
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P mod 1000000007。
数据范围:对于 50%的数据, size≤10^4。
对于100%的数据, size≤10^5。
数组中所有数字的值满足 0≤val≤10^9。
要求:空间复杂度O(n),时间复杂度O(nlogn)。
输入描述:题目保证输入的数组中没有的相同的数字。
示例1
输入:[1,2,3,4,5,6,7,0]
返回值:7
示例2
输入:[1,2,3]
返回值:0
思路:归并排序,递归
代码:
public class Solution {
public int mod = 1000000007;
public int mergeSort(int left, int right, int[] data, int [] temp){
//终止条件,停止划分
if(left >= right){
return 0;
}
//取中间,将数组不断进行拆分
int mid = (left + right) >> 1;
//递归分别求出左半区间和右半区间的逆序对个数(左半区间和右半区间都是有序的)进行左右划分合并
int res = mergeSort(left, mid, data, temp) + mergeSort(mid + 1, right, data, temp);
//防止溢出
res %= mod;
//i是第一个有序小数组(左半区间)的起始索引,j是第二个有序小数组(右半区间)的起始索引
int i = left, j = mid + 1;
//将原数组data中的元素拷贝到temp数组中
for(int k = left; k <= right; k++){
temp[k] = data[k];
}
//开始合并,k表示当前处理到原数组的哪个元素
for(int k = left; k <= right; k++){
if(i == mid + 1) {
//说明左半区间已经全部处理完毕,将右半区间的所有值写回原数组
data[k] = temp[j++];
}else if(j == right + 1 || temp[i] <= temp[j]){
//说明右半区间已经全部处理完毕 或者 左半区间元素 <= 右半区间元素,此时不构成逆序对
data[k] = temp[i++];
}else{
//说明左半区间元素 > 右半区间元素,此时构成逆序对
data[k] = temp[j++];
//此时i和j对应的区间中逆序对的个数 = (mid - i + 1)个
res += mid - i + 1;
}
}
return res % mod;
}
public int InversePairs(int [] array) {
int n = array.length;
int[] res = new int[n];
return mergeSort(0, n - 1, array, res);
}
}