數列上的分治(求逆序數)

在網上看,逆序數可以有線段樹解決,我不太清楚,不過最近我在學習如何用分治處理逆序數問題。

大家看看代碼吧,看完估計就懂了(如果還不懂去看下分治排序,逆序數問題是在其基礎之上的)


<span style="font-size:18px;">#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
vector<int> A;
int merge_count(vector<int> &a)
{
    int n=a.size();
    if(n<=1) return 0;

    int cnt=0;
    vector<int> b(a.begin(),a.begin()+n/2);
    vector<int> c(a.begin()+n/2,a.end());

    cnt+=merge_count(b);
    cnt+=merge_count(c);

    int ai=0,bi=0,ci=0;
    while(ai<n){
        if(bi<b.size() && (ci==c.size() || b[bi]<=c[ci])){
            a[ai++]=b[bi++];
        }
        else{
            cnt+=n/2-bi; /// 求逆序數只要在分治排序上添加這個語句就好
            a[ai++]=c[ci++];
        }
    }
    return cnt;
}
int main()
{
    int n,x;
    scanf("%d",&n);
    for(int i=0;i<n;i++){
        scanf("%d",&x);
        A.push_back(x);
    }
    printf("%d\n",merge_count(A)); /// 逆序數
    vector<int>::iterator it=A.begin();
    while(it!=A.end())
    {
        printf("%d%c",*it,(it!=(A.end()-1)?' ':'\n')); /// 分治排序結果
        it++;
    }
    return 0;
}
</span>


發佈了106 篇原創文章 · 獲贊 5 · 訪問量 5萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章