原题目:
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总> 数P。并将P对1000000007取模的结果输出。 即输出P%1000000007
首先拿到该题最直观的做法是利用每一个数,与其后的另外一个数作比较,如果后面一个数比前面一个数小,则构成逆序对,简单明了,但是比较的次数较多,平均每个数需要比较n次,总共的时间复杂度为 O ( n 2 ) O(n^2) O(n2),一旦数据较多,必然会超时。因此考虑别的方法。
可以利用分治的方法来处理,将整体问题局部化。
思考一下,为什么这里最后最后看上去像是一个排序的问题,因为有序的数组不存在逆序对问题。
下面结合上面的归并排序,分析一下,为什么可以将归并排序利用到逆序对个数的求解中。
比如针对于中间各自有两个元素的子数组:
第一步:分别指向两个子数组末端,并且将复制数组下标指向最末端。
第二步:比较p1及P2下标指向数组的值,如果num[p1]<num[p2],则说明两数不构成逆序对,直接将num[p2]复制到备用数组中;如果num[p1]>num[p2],则说明这两个数组是一个逆序对;并且p2以前的数据与num[p1]都构成逆序对。(因为右边数组肯定是已经排序好的)。
第三步:继续比较num[p1]与num[p2]的大小,由于num[p1]>num[p2],因此两数构成一个逆序对,p1左移,并且将大值复制到备用数组中。
第四步:继续比较,直到p1或者p2小于两个子数组任意左边界,然后将剩余的数据拷贝到备用数组中,备用数组中的数据已经是有序的。
第五步:不断的重复以上步骤,直到p1=p2。
基于以上思想,代码如下:
public class Solution {
public int InversePairs(int [] array) {
if(array == null || array.length <= 0){
return 0;
}
int length = array.length;
// 创建一个备用数组
int[] copy = new int[length];
int result = mergeNum(array,copy,0,length - 1);
return result;
}
public int mergeNum(int[] num,int[] copy,int start,int end){
// 首先考虑什么时候不分数组了
if(start == end){
return 0;
}
// 两个子数组的分界下标
int middle = (start + end)/2;
// 左子数组求得的逆序对数
int left = mergeNum(num,copy,start,middle)%1000000007;
// 右子数组求得的逆序对数
int right = mergeNum(num,copy,middle+1,end)%1000000007;
int i = middle;
int j = end;
// 逆序对的个数
int count = 0;
// 备用数组的下标
int indexCopy = end;
// 下面是真正的归并过程
while(i >= start && j >= middle + 1){
if(num[i] > num[j]){
copy[indexCopy--] = num[i--];
count += j - (middle);
}else{
copy[indexCopy--] = num[j--];
}
}
// 上述循环完成之后如果两个子数组都有没有遍历完的数,则继续遍历各个子数组
for(; i >= start ;i--){
copy[indexCopy--] = num[i];
}
for(; j >= middle + 1;j--){
copy[indexCopy--] = num[j];
}
// 为什么有这一步?将子数组变成有序的。
for(int k = start;k <= end;k++){
num[k] = copy[k];
}
return (left + right + count)%1000000007;
}
}
上面的解答是第一版本,只是通过的50%,因为题目中说将P对1000000007取模的结果输出。 即输出P%1000000007
,则需要对每次count求余数。
因此需要将这段代码改成如下形式:
// 下面是真正的归并过程
while(i >= start && j >= middle + 1){
if(num[i] > num[j]){
copy[indexCopy--] = num[i--];
count += j - (middle);
if(count>=1000000007){
count%=1000000007;
}
}else{
copy[indexCopy--] = num[j--];
}
}
总结
该题整体的思想就是归并排序,先分、再归并。类似的还有快排。