1. 程式人生 > >HDU 2689 Sort it 歸併排序求逆序對

HDU 2689 Sort it 歸併排序求逆序對

題目連結:http://acm.hdu.edu.cn/showproblem.php?pid=2689

思路:就是歸併排序啦。在歸併排序的同時,記錄逆序對就行了。時間複雜度即為歸併排序的複雜度:O(N log N)。

程式碼:

#include<iostream>
#include<cstdio>
#define maxn 100010

using namespace std;

int N,A[maxn],T[maxn],ans;

void merge_sort(int L,int R){
    if(R-L<=0) return;
    int m=(L+R)/2;
    merge_sort(L,m);merge_sort(m+1,R);//遞迴歸併排序
    int i=L,j=m+1,cnt=0;
    while(i<=m || j<=R)
        if(i>m || (j<=R && A[j]<A[i])) T[cnt++]=A[j++];
        else {
            if(i<=m) ans+=(j-m-1);//記錄逆序對數
            T[cnt++]=A[i++];
        }
    for(i=0;i<cnt;i++) A[i+L]=T[i];
}

int main(){
    while(cin>>N){
        for(int i=0;i<N;i++) scanf("%d",A+i);
        ans=0;
        merge_sort(0,N-1);
        printf("%d\n",ans);
    }
    return 0;
}