BZOJ 2141: 排队|树状数组套主席树

—————做的第一道树套树的题—————

杂乱程度为逆序对的数量（i< j并且a[i]>a[j]）
可以发现交换l和r对答案的影响只与区间（l，r）的数有关
交换l和r对答案作出的改变为：
加上（l，r）中大于a[l]的数的个数和小于a[r]的数的个数
减去（l，r）中小于a[l]的数的个数和大于a[r]的数的个数
还有a[l],和a[r]本身对逆序对的贡献
带有区间修改主席树需要套上一层树状数组
似乎正确的姿势是分块？

#include<set>
#include<map>
#include<ctime>
#include<queue>
#include<cmath>
#include<cstdio>
#include<vector>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define lowbit(x) (x&(-x))
#define MX 1e9
#define pa pair<int,int>
using namespace std;
int sc()
{
    int i=0,f=1; char c=getchar();
    while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
    return i*f;
}
struct W{int x,pos;}q[20002];
int a[20002],tr[20002];
int root[20002],ch[6000002][2],v[6000002];
int cnt,n,m,mx,ans=0;
bool cmp(W a,W b){return a.x<b.x;}
void add(int &pre,int &x,int l,int r,int u,int f)
{
    if(!x)x=++cnt;  v[x]=v[pre]+f; 
    if(l==r)return;
    int mid=l+r>>1;
    if(u<=mid)
    {
        ch[x][1]=ch[pre][1];
        int y=ch[pre][0];   ch[x][0]=++cnt;
        add(y,ch[x][0],l,mid,u,f);
    }
    else
    {
        ch[x][0]=ch[pre][0];
        int y=ch[pre][1];   ch[x][1]=++cnt;
        add(y,ch[x][1],mid+1,r,u,f);
    }
}
int _ask(int x,int l,int r,int w)
{
    if(r<=w)return v[x];
    int mid=l+r>>1;
    if(mid>=w)return _ask(ch[x][0],l,mid,w);
    return v[ch[x][0]]+_ask(ch[x][1],mid+1,r,w);
}
int ask(int pos,int x)
{
    if(x==0)return 0;   int ans=0;
    for(int i=pos;i;i-=lowbit(i))   ans+=_ask(root[i],1,mx,x);
    return ans;
}
void change(int x,int u,int f)
{
    for(int i=x;i<=n;i+=lowbit(i))
        add(root[i],root[i],1,mx,u,f);
}
int main()
{
    n=sc();
    for(int i=1;i<=n;i++)q[i]=(W){sc(),i};
    sort(q+1,q+n+1,cmp);
    for(int i=1;i<=n;i++)
        if(q[i].x!=q[i-1].x)
            a[q[i].pos]=++mx;
        else 
            a[q[i].pos]=mx;
    for(int i=n;i>=1;i--)
    {
        int x=a[i];
        for(int j=x-1;j;j-=(j&(-j)))    ans+=tr[j];
        for(int j=x;j<=mx;j+=(j&(-j)))    tr[j]+=1;
    }
    for(int i=1;i<=n;i++) change(i,a[i],1);
    printf("%d\n",ans);
    m=sc();
    for(int i=1;i<=m;i++)
    {
        int l=sc(),r=sc();  
        change(l,a[l],-1);
        change(r,a[r],-1);
        change(r,a[l],1);
        change(l,a[r],1);
        int a1=ask(r-1,a[r]-1)-ask(l,a[r]-1); 
        int a2=ask(r-1,a[l]-1)-ask(l,a[l]-1); 
        int b1=r-l-1-(ask(r-1,a[r])-ask(l,a[r]));
        int b2=r-l-1-(ask(r-1,a[l])-ask(l,a[l]));
        swap(a[l],a[r]);
        if(a[l]>a[r])ans++;else if(a[l]<a[r])ans--;
        ans+=a1-b1+b2-a2;
        printf("%d\n",ans);
    }
    return 0;
}