题目描述
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007
题目链接:牛客网
输入描述:
题目保证输入的数组中没有的相同的数字
数据范围:
对于%50的数据,size<=10^4
对于%75的数据,size<=10^5
对于%100的数据,size<=2*10^5
解题思路
public class Main {
public static long cnt = 0;
public static int[] temp; // 在这里声明辅助数组,而不是在 merge() 递归函数中声明
public static void main(String[] args) {
int[] data = {1,2,3,4,5,6,7,0};
System.out.println(inversePairs(data));
}
public static int inversePairs(int[] nums) {
temp = new int[nums.length];
mergeSort(nums,0,nums.length - 1);
return (int)(cnt % 1000000007);
}
public static void mergeSort(int[] nums,int l,int h) {
if (h - l < 1) {
return;
}
int m = l + (h - l) / 2;
mergeSort(nums, l, m);
mergeSort(nums, m + 1, h);
merge(nums, l, m, h);
}
public static void merge(int[] nums,int l,int m,int h) {
int i = l,j = m + 1,k = l;
while (i <= m || j <= h) {
if (i > m) {
temp[k] = nums[j++];
}else if (j > h) {
temp[k] = nums[i++];
} else if (nums[i] < nums[j]) {
temp[k] = nums[i++];
} else {
temp[k] = nums[j++];
cnt += m - i + 1; // nums[i] > nums[j],说明 nums[i...mid] 都大于 nums[j]
}
k++;
}
for (k = l;k <= h;k++) {
nums[k] = temp[k];
}
}
}