codeforeces 540E(樹狀數組

標籤：

題意：交換自然數中的若干對數，求交換後總共有多少逆序數對。

思路：因為題目數字範圍比較大，不能直接用樹狀數組算，首先要離散化。然後一種演算法是官方題解中根據逆序對數是否屬於交換過的數分類討論統計。我的演算法是把沒有交換的連續的數看成一個數，使用樹狀數組統計的時候直接加上這個區間的數字個數，這樣就不需要繁瑣（？）的討論了。

#include<iostream>#include<map>#include<algorithm>#include<cstdio>#include<cstring>#include<cstdlib>#include<vector>#include<queue>#include<stack>#include<functional>#include<set>#define pb push_back#define fs first#define se secondusing namespace std;typedef long long ll;typedef pair<ll,ll> P;const int maxv=1e5*4;struct BIT{    ll a[maxv];    void add(int p,int x){        while(p<maxv){            a[p]+=x;            p+=p&-p;        }    }    ll sum(int p){        ll ans=0;        while(p>0){            ans+=a[p];            p-=p&-p;        }        return ans;    }    ll after(int a){        return sum(maxv-1)-sum(a);    }}B;int a[maxv];int b[maxv];int c[maxv];int mid[maxv];int sq[maxv];int n;map<int,int> mii;int main(){    freopen("/home/files/CppFiles/in","r",stdin);    cin>>n;    for(int i=0;i<n;i++){        scanf("%d%d",a+i,b+i);        c[i*2]=a[i];        c[i*2+1]=b[i];    }    sort(c,c+2*n);    int h=unique(c,c+2*n)-c;    for(int i=1;i<h;i++){        mid[i]=c[i]-c[i-1]-1;    }    for(int i=0;i<h;i++){        mii[c[i]]=i+1;    }    for(int i=0;i<maxv;i++){        sq[i]=i;    }    for(int i=0;i<n;i++){        swap(sq[mii[a[i]]],sq[mii[b[i]]]);    }    ll ans=0;    for(int i=1;i<=h;i++){        ll now=sq[i];        ans+=B.after(now*2-1);        B.add(now*2-1,1);        ans+=mid[i]*B.after(i*2);        B.add(2*i,mid[i]);    }    cout<<ans<<endl;    return 0;}

View Code

 

codeforeces 540E(樹狀數組