LUOGU P3157 [CQOI2011]动态逆序对(CDQ 分治)

传送门

解题思路

cdq分治，将位置看做一维，修改时间看做一维，权值看做一维，然后就转化成了三维偏序，用排序+cdq+树状数组。注意算删除贡献时要做两次cdq，分别算对前面和后面的贡献。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;
const int MAXN = ;
const int MAXM = ;
const int MAXQ = MAXN+MAXM;
typedef long long LL;

inline int rd(){
    int x=,f=;char ch=getchar();
    while(!isdigit(ch)) {f=ch=='-'?:;ch=getchar();}
    while(isdigit(ch))  {x=(x<<)+(x<<)+ch-'';ch=getchar();}
    return f?x:-x;
}

LL ans,f[MAXN];
int n,m,now;

struct Query{
    int t,pos,id,a;
    LL pre,nxt;
}q[MAXN],tmp[MAXN];

void add(int x,int y){
    for(;x<=n;x+=x&-x) f[x]+=y;
}

LL query(int x){
    LL ret=;
    for(;x;x-=x&-x) ret+=f[x];
    return ret;
}

void Clear(int x){
    for(;x<=n;x+=x&-x) f[x]=;
}

inline bool cmp(Query A,Query B){
    return A.a>B.a;
}

inline bool _cmp(Query A,Query B){
    return A.a<B.a;
}

inline bool cmp_(Query A,Query B){
    return A.pos<B.pos;
}

inline bool cmp1(Query A,Query B){
    return A.t>B.t;
}

inline bool cmp2(Query A,Query B){
    return A.t<B.t;
}

void cdq(int l,int r){
    if(l==r) return;
    int mid=l+r>>;cdq(l,mid);cdq(mid+,r);
    int L=l,R=mid+,o=;
    while(L<=mid && R<=r){
        if(q[L].pos>q[R].pos){
            add(q[L].a,);
            tmp[++o]=q[L++];
        }
        else{
            q[R].pre+=query(q[R].a);
            tmp[++o]=q[R++];
        }
    }
    while(L<=mid) tmp[++o]=q[L++];
    while(R<=r){
        q[R].pre+=query(q[R].a);
        tmp[++o]=q[R++];
    }
    for(register int i=l;i<=mid;i++) Clear(q[i].a);
    for(register int i=;i<=o;i++) q[i+l-]=tmp[i];
}

void CDQ(int l,int r){
    if(l==r) return;
    int mid=l+r>>;CDQ(l,mid);CDQ(mid+,r);
    int L=l,R=mid+,o=;
    while(L<=mid && R<=r){
        if(q[L].a>q[R].a){
            add(q[L].pos,);
            tmp[++o]=q[L++];
        }
        else{
            q[R].nxt+=query(q[R].pos);
            tmp[++o]=q[R++];
        }
    }
    while(L<=mid) tmp[++o]=q[L++];
    while(R<=r){
        q[R].nxt+=query(q[R].pos);
        tmp[++o]=q[R++];
    }
    for(register int i=l;i<=mid;i++) Clear(q[i].pos);
    for(register int i=;i<=o;i++) q[i+l-]=tmp[i];
}

int main(){
    n=rd(),m=rd();int x;
    for(int i=;i<=n;i++)
        q[i].a=rd(),q[i].pos=i;
    sort(q+,q++n,cmp);
    for(int i=;i<=n;i++) {
        add(q[i].pos,);
        ans+=query(q[i].pos-);
    }
    for(int i=;i<=n;i++) add(q[i].pos,-);
    sort(q+,q++n,_cmp);
//    for(int i=1;i<=n;i++) cout<<q[i].pos<<" ";
//    cout<<ans<<endl;
    for(int i=;i<=m;i++) q[rd()].t=i;
    sort(q+,q++n,cmp_);
    for(int i=;i<=n;i++) if(q[i].t==) q[i].t=++now+m;
    sort(q+,q++n,cmp1);
    cdq(,n);
    sort(q+,q++n,cmp1);
    CDQ(,n);
    sort(q+,q++n,cmp2);
    for(int i=;i<=m;i++) {
        printf("%lld\n",ans);
        ans-=q[i].pre+q[i].nxt;
    }
    return ;
}