bzoj3295: [Cqoi2011]动态逆序对

传送门
树状数组套主席树。
外层树状数组记录区间
内层主席树记录数字出现个数
删减时枚举在前面比他小的删去的点的数量
和在后面比他大的删去的点的数量
求值可以用树状数组实现。
时间复杂度nlogn^2

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
#define inf 0x3f3f3f3f
#define ll long long
#define N 100005
#define M 5000005
using namespace std;
ll ans;
int n,m,sz,x;
int a[35],b[35];
int num[N],pos[N],a1[N],a2[N],t[N],rt[N];
int ls[M],rs[M],sum[M];
inline int ask(int x){
    int s=0;
    for (int i=x;i;i-=i&(-i))
        s+=t[i];
    return s;
}
void change(int &y,int l,int r,int x){
    if (!y) y=++sz;
    sum[y]++;
    if (l==r) return;
    int mid=(l+r)/2;
    if (x<=mid) change(ls[y],l,mid,x);
    else change(rs[y],mid+1,r,x);
}
int askmore(int x,int y,int num){
    a[0]=b[0]=0;
    int tmp=0;
    for (int i=x-1;i;i-=i&(-i))
        a[++a[0]]=rt[i];
    for (int i=y;i;i-=i&(-i))
        b[++b[0]]=rt[i];
    int l=1,r=n;
    while (l<r){
        int mid=(l+r)/2;
        if (num<=mid){
            for (int i=1;i<=a[0];i++) tmp-=sum[rs[a[i]]];
            for (int i=1;i<=b[0];i++) tmp+=sum[rs[b[i]]];
            for (int i=1;i<=a[0];i++) a[i]=ls[a[i]];
            for (int i=1;i<=b[0];i++) b[i]=ls[b[i]];
            r=mid;
        }
        else{
            for (int i=1;i<=a[0];i++) a[i]=rs[a[i]];
            for (int i=1;i<=b[0];i++) b[i]=rs[b[i]];
            l=mid+1;
        }
    }
    return tmp;
}
int askless(int x,int y,int num){
    a[0]=b[0]=0;
    int tmp=0;
    for (int i=x-1;i;i-=i&(-i))
        a[++a[0]]=rt[i];
    for (int i=y;i;i-=i&(-i))
        b[++b[0]]=rt[i];
    int l=1,r=n;
    while (l<r){
        int mid=(l+r)/2;
        if (num>mid){
            for (int i=1;i<=a[0];i++) tmp-=sum[ls[a[i]]];
            for (int i=1;i<=b[0];i++) tmp+=sum[ls[b[i]]];
            for (int i=1;i<=a[0];i++) a[i]=rs[a[i]];
            for (int i=1;i<=b[0];i++) b[i]=rs[b[i]];
            l=mid+1;
        }
        else{
            for (int i=1;i<=a[0];i++) a[i]=ls[a[i]];
            for (int i=1;i<=b[0];i++) b[i]=ls[b[i]];
            r=mid;
        }
    }
    return tmp;
}
int main(){
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++){
        scanf("%d",&num[i]);
        pos[num[i]]=i;
        a1[i]=ask(n)-ask(num[i]);
        ans+=a1[i];
        for (int j=num[i];j<=n;j+=j&(-j))
            t[j]++;
    }
    memset(t,0,sizeof(t));
    for (int i=n;i>=1;i--){
        a2[i]=ask(num[i]-1);
        for (int j=num[i];j<=n;j+=j&(-j))
            t[j]++; 
    }
    for (int i=1;i<=m;i++){
        printf("%lld\n",ans);
        scanf("%d",&x); x=pos[x];
        ans-=(a1[x]+a2[x]-askmore(1,x-1,num[x])-askless(x+1,n,num[x]));
        for (int j=x;j<=n;j+=j&(-j))
            change(rt[j],1,n,num[x]);
    }
}