bzoj:2141: 排队

Description

排排坐，吃果果，生果甜嗦嗦，大家笑呵呵。你一个，我一个，大的分给你，小的留给我，吃完果果唱支歌，大家乐和和。红星幼儿园的小朋友们排起了长长地队伍，准备吃果果。不过因为小朋友们的身高有所区别，排成的队伍高低错乱，极不美观。设第i个小朋友的身高为hi，我们定义一个序列的杂乱程度为：满足ihj的(i,j)数量。幼儿园阿姨每次会选出两个小朋友，交换他们的位置，请你帮忙计算出每次交换后，序列的杂乱程度。为方便幼儿园阿姨统计，在未进行任何交换操作时，你也应该输出该序列的杂乱程度。

Input

第一行为一个正整数n，表示小朋友的数量；第二行包含n个由空格分隔的正整数h1,h2,…,hn，依次表示初始队列中小朋友的身高；第三行为一个正整数m，表示交换操作的次数；以下m行每行包含两个正整数ai和bi¬，表示交换位置ai与位置bi的小朋友。

Output

输出文件共m行，第i行一个正整数表示交换操作i结束后，序列的杂乱程度。

Sample Input

【样例输入】
3
130 150 140
2
2 3
1 3

Sample Output

1
0
3

 

 

树状数组套treap，挺好理解的……

只是有点慢……

#include<cstdio>
#include<cstdlib>
#include<algorithm>
using namespace std;

int n,m,l,r,ans=0,o,p,num=0;
inline int read(){
    p=0;o=getchar();
    while(o<'0'||o>'9') o=getchar();
    while(o>='0'&&o<='9') p=p*10+o-48,o=getchar();
    return p;
}
int root[20001],a[20001];
struct tree{
    int l,r,k,ra,w,f,s;
    tree(){
        s=f=l=r=0;
    }
};
tree t[2000000];
inline void ler(int &p){
    int k=t[p].r;
    t[p].r=t[k].l;
    t[k].l=p;
    t[k].s=t[p].s;
    t[p].s=t[t[p].l].s+t[t[p].r].s+t[p].w;
    p=k;
}
inline void rir(int &p){
    int k=t[p].l;
    t[p].l=t[k].r;
    t[k].r=p;
    t[k].s=t[p].s;
    t[p].s=t[t[p].l].s+t[t[p].r].s+t[p].w;
    p=k;
}
inline void insert(int &p,int k){
    if (p==0){
        p=++num;
        t[p].k=k;
        t[p].w=1;
        t[p].s=1;
        t[p].ra=rand();
        return;
    }
    t[p].s++;
    if (t[p].k==k) t[p].w++;else
    if (t[p].k>k){
        insert(t[p].l,k);
        if (t[t[p].l].ra<t[p].ra) rir(p);
    }else{
        insert(t[p].r,k);
        if (t[t[p].r].ra<t[p].ra) ler(p);
    }
}
inline void dell(int &p){
    if (t[p].l==0&&t[p].r==0) p=0;else
    if (t[p].l==0) ler(p),dell(t[p].l);else
    if (t[p].r==0) rir(p),dell(t[p].r);else
    if (t[t[p].l].ra<t[t[p].r].ra) rir(p),dell(t[p].r);else ler(p),dell(t[p].l);
}
inline void del(int &p,int k){
    if (p==0) return;
    t[p].s--;
    if (k==t[p].k){
        t[p].w--;
        if (!t[p].w) dell(p);
        return;
    }
    if (k<t[p].k) del(t[p].l,k);else del(t[p].r,k);
}
inline int qua(int p,int k){
    if (p==0) return 0;
    if (t[p].k==k) return t[t[p].r].s;else
    if (t[p].k>k) return qua(t[p].l,k)+t[p].w+t[t[p].r].s;else return qua(t[p].r,k);
}
inline int qui(int p,int k){
    if (p==0) return 0;
    if (t[p].k==k) return t[t[p].l].s;else
    if (t[p].k>k) return qui(t[p].l,k);else return qui(t[p].r,k)+t[p].w+t[t[p].l].s;
}
inline int lo(int x){return x&(-x);}
inline void in(int i,int k){
    while(i<=n){
        insert(root[i],k);
        i+=lo(i);
    }
}
inline void de(int i,int k){
    while(i<=n){
        del(root[i],k);
        i+=lo(i);
    }
}
inline int ask(int x,int k){
    int s=0;
    while(x>0){
        s+=qua(root[x],k);
        x-=lo(x);
    }
    return s;
}
inline int aski(int x,int k){
    int s=0;
    while(x>0){
        s+=qui(root[x],k);
        x-=lo(x);
    }
    return s;
}
int main(){
    register int i,j;
    n=read();
    for (i=1;i<=n;i++) in(i,a[i]=read());
    for (i=1;i<=n;i++) ans+=ask(i,a[i]);
    m=read();
    printf("%d\n",ans);
    while(m--){
        l=read();r=read();
        if (r<l) swap(r,l);
        ans+=(ask(r-1,a[l])-ask(l,a[l]))-(aski(r-1,a[l])-aski(l,a[l]))+(aski(r-1,a[r])-aski(l,a[r]))-((ask(r-1,a[r])-ask(l,a[r])));
        if (a[l]>a[r]) ans--;else if (a[l]<a[r]) ans++;
        printf("%d\n",ans);
        if (!m) return 0;
        de(l,a[l]);de(r,a[r]);
        swap(a[l],a[r]);
        in(l,a[l]);in(r,a[r]);
    }
}