在數組中的兩個數字,如果前面一個數字大于后面的數字,則這兩個數字組成一個逆序對。輸入一個數組,求出這個數組中的逆序對的總數P。并將P對1000000007取模的結果輸出。 即輸出P%1000000007
public class Solution {
public int InversePairs(int [] array) {
int len = array.length;
return (int) dfs(array, 0, len-1)%1000000007; // 類似歸并排序
}
public double dfs(int[] array, int start, int end){
if(start>=end)
return 0;
int mid = (start + end)/2;//分成兩段
double leftPartCnts = dfs(array,start,mid);//左段內部的逆序數
double rightPartCnts = dfs(array, mid+1, end);//右段的逆序數
int foreIdx = mid;//左段的指針,初始指向最后一個元素
int endIdx = end;//右段的指針,初始指向最后一個元素
long cnts = 0;//記錄合并產生的逆序數
int[] copy = new int[end-start+1];//合并時用到的臨時數組
int copyIdx = copy.length-1;//從右往左填充臨時數組
while(foreIdx >= start && endIdx >= mid+1){//如果左段的元素和右段的元素都沒有遍歷完
if(array[foreIdx]>array[endIdx]){// 如果左段元素 > 右段元素
copy[copyIdx--] = array[foreIdx--]; //把較大的元素放入copy
cnts += endIdx-mid; // 此時endIdx前的元素和foreIdx肯定也是逆序對,所以新增了endIdx-(mid+1—)+1個逆序對
}else{
copy[copyIdx--] = array[endIdx--]; //不產生逆序對,直接放入copy
}
}
while(foreIdx>=start){
copy[copyIdx--] = array[foreIdx--];// 把左段剩余的元素加入copy
}
while(endIdx>=mid+1){
copy[copyIdx--] = array[endIdx--];// 把右段剩余的元素加入copy
}
int tmpStart =start;
for(int el:copy){ // 把copy放到array對應的位置
array[tmpStart++] = el;
}
if((leftPartCnts + rightPartCnts + cnts)>1000000007){//如果超出1000000007,先取余不影響最終結果,同時避免溢出
return (leftPartCnts + rightPartCnts + cnts)%1000000007;
}
else return leftPartCnts + rightPartCnts + cnts;//逆序數=兩斷數組內部的逆序數+兩段數組合并產生的逆序數
}
}