【Bzoj2141】排队

题意

给一个序列，每次可以交换其中的两个数，每次交换后输出逆序对的数量。

---

解析

每次交换考虑x,y之间的数字(x之前,y之后的数字是没有影响的)
那么增加的数字是小于a[y],大于a[x]的那些。
减小的是小于a[x],大于a[y]的。
区间查rank显然可以树套树解决。
然后注意一些特判。

---

#include <cstdio>
#include <algorithm>

#define Rep( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i<=(i##_END);i++)
#define For( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i!=(i##_END);i++)
#define Lop( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i>=(i##_END);i--)
#define Dnt( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i!=(i##_END);i--)

using std :: max;
using std :: min;

const int maxx = 20000 + 25;
const int maxm = 20000 * 16 * 2 + 25;

typedef int Array[maxm];

int n,m,x,y,z,num,f,tot,ans;
int a[maxx],rt[maxx],T[maxx];
int ch[maxm][2];

Array size,cnt,v,fix;

namespace Treap{

    void upt(int i) {size[i] = size[ch[i][0]] + size[ch[i][1]] + cnt[i];}

    int chk(int x,int y) {return x == y? 20011025 : x > y;}

    void rotate(int &i,int f){
        int t = ch[i][f];
        ch[i][f] = ch[t][f^1];
        ch[t][f^1] = i;
        size[t] = size[i];
        upt(i);i = t;
    }

    void insert(int &i,int x){
        if(!i) {v[i=++num] = x;size[i] = cnt[i] = 1;fix[i] = rand();return;}
        size[i] ++;
        if(x == v[i]) {cnt[i] ++;return;}
        int tmp = chk(x,v[i]);
        insert(ch[i][tmp],x);
        if(fix[ch[i][tmp]] < fix[i]) rotate(i,tmp);
    }

    void remove(int &i,int x){
        if(!i) return;
        int tmp = chk(x,v[i]);
        if(x != v[i]) {size[i] --;remove(ch[i][tmp],x);}
        if(x == v[i]){
            if(cnt[i] > 1) {cnt[i] --;size[i] --;return;}
            if(ch[i][0] * ch[i][1] == 0) {i = ch[i][0] + ch[i][1];return;}
            tmp = chk(fix[ch[i][0]],fix[ch[i][1]]);
            rotate(i,tmp);remove(i,x);
        }
    }

    int Qrank(int i,int x){
        if(!i) return 0;
        if(v[i] == x) return size[ch[i][0]];
        else return x <= v[i]? Qrank(ch[i][0],x) : Qrank(ch[i][1],x) + cnt[i] + size[ch[i][0]];
    }

    int faq(int i,int x){
        if(!i) return 0;
        if(v[i] == x) return cnt[i];
        else return x <= v[i]? faq(ch[i][0],x) : faq(ch[i][1],x);
    }

}

namespace Bit{

    using namespace Treap;

    void Add(int x,int k,int f){
        for(int i=x;i<=n;i+=i&(-i))
            if(f == 1) insert(rt[i],k);
            else remove(rt[i],k);

    }

    int Query(int x,int k,int f){
        int ans = 0;
        for(int i=x;i;i-=i&(-i))
            ans += f? Qrank(rt[i],k) : faq(rt[i],k);
        return ans;
    }

}

using namespace Bit;

namespace Initialization{

    int main(){
        Lop( i , n , 1 ){
            Add(i,a[i],1);
            ans += (Query(n,a[i],1) - Query(i,a[i],1));
        }
    }

}

int main(){
    scanf("%d",&n);
    Rep( i , 1 , n ) scanf("%d",&a[i]);
    Initialization :: main();
    printf("%d\n",ans);
    scanf("%d",&m);
    while( m-- ){
        scanf("%d%d",&x,&y);
        if(x > y) std :: swap(x,y);
        if(x == y || a[x] == a[y]) {printf("%d\n",ans);continue;}
        int tmp1 = Query(y-1,a[x],1) - Query(x,a[x],1);
        int tmp2 = Query(y-1,a[y],1) - Query(x,a[y],1);
        int tmp3 = (y-x-1) - (Query(y-1,a[y],1) - Query(x,a[y],1)) - (Query(y-1,a[y],0) - Query(x,a[y],0));
        int tmp4 = (y-x-1) - (Query(y-1,a[x],1) - Query(x,a[x],1)) - (Query(y-1,a[x],0) - Query(x,a[x],0));

        ans = ans - tmp1 + tmp2 - tmp3 + tmp4;
        ans += (a[x]<a[y])? 1 : -1;

        Add(x,a[x],2);Add(y,a[x],1);
        Add(y,a[y],2);Add(x,a[y],1);
        std :: swap(a[x],a[y]);

        printf("%d\n",ans);
    }
    return 0;
}