遞歸和分治:poj2299

/*
總體思想使用分治法,進行歸併排序,求出每次合併的逆序總數累加。說明一下:按照算導中方法寫的程序
有可能會超時,因爲每次merge都重新分配兩個數組,且需單獨循環計算逆序,操作較多。另外,最後結果sum使用long long,
在最壞情況下是n^2級的,n<500000,一個long(4B,2^32) 不夠 。 3692K 375MS
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

const int N = 500005;
long long sum;
int vec[N],back[N];

void merge(int s,int m,int e){
    int i = s, j = m + 1, k = s;
    while (i <= m && j <= e) {
        if (vec[i] > vec[j]) {
            sum += m - i + 1;   //逆序數累加 
            back[k++] = vec[j++];
        } else {
            back[k++] = vec[i++];
        }
    }
    while (i <= m)
        back[k++] = vec[i++];
    while (j <= e)
        back[k++] = vec[j++];
    memcpy(vec + s, back + s, sizeof(*back) * (e - s + 1));
}

void merge_sort(int p,int r){
    if (p<r){
        int q=(p+r)/2;
        merge_sort(p,q);
        merge_sort(q+1,r);
        merge(p,q,r);
    }
}      
int main(){
    int i,n;
    scanf("%d",&n);
    while(n!=0){
        sum = 0;
        for(i=1;i<=n;i++){
            scanf("%d",vec+i);
        }
        merge_sort(1,n);
        printf("%lld\n",sum);
        scanf("%d",&n);
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章