剑指Offer-51. 数组中的逆序对
程序员文章站
2022-03-05 13:30:00
...
题目描述
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007
输入描述:
题目保证输入的数组中没有的相同的数字
数据范围:
对于%50的数据,size<=10^4
对于%75的数据,size<=10^5
对于%100的数据,size<=2*10^5
思路
利用归并排序,在合并时同时计算逆序个数。
public class Solution {
private long count = 0;
private int[] tmp;
public int InversePairs(int [] array) {
int n = array.length;
tmp = new int[n];
mergeSort(array, 0, n - 1);
return (int)count % 1000000007;
}
public void mergeSort(int[] arr, int left, int right) {
if (right - left < 1) return;
int mid = left + (right - left) / 2;
mergeSort(arr, left, mid);
mergeSort(arr, mid + 1, right);
merge(arr, left, mid, right);
}
public void merge(int[] arr, int left, int mid, int right) {
int i = left;
int j = mid + 1;
int t = 0;
while (i <= mid && j <= right) {
if (arr[i] > arr[j]) {
count += (mid - i + 1);
if (count > 1000000007)
count %= 1000000007;
tmp[t++] = arr[j++];
} else {
tmp[t++] = arr[i++];
}
}
while (i <= mid) tmp[t++] = arr[i++];
while (j <= right) tmp[t++] = arr[j++];
int k = 0;
while (left <= right) {
arr[left++] = tmp[k++];
}
}
}