BZOJ 3295: [Cqoi2011]动态逆序对 [CDQ分治]

首先可以看成倒着插入，求逆序对数

每个数分配时间(注意每个数都要一个时间)$t$，$x$位置，$y$数值

$CDQ(l,r)$时归并排序$x$

然后用$[l,mid]$的加入更新$[mid+1,r]$的查询(其实每个数就是一个插入一个查询)

这里就是前后求逆序对，用树状数组

#include <iostream>

#include <cstdio>

#include <algorithm>

#include <cstring>

using namespace std;

typedef long long ll;

const int N=2e5+;

inline int read(){

    char c=getchar();int x=,f=;

    while(c<''||c>''){if(c=='-')f=-;c=getchar();}

    while(c>=''&&c<=''){x=x*+c-'';c=getchar();}

    return x*f;

}

int n,m;

int mp[N];

struct Operation{

    int t,x,y;

    Operation(){}

    Operation(int t,int id,int v):t(t),x(id),y(v){}

    bool operator <(const Operation &r)const{

        return x==r.x ? y<r.y : x<r.x;

    }

}a[N],t[N];

inline bool cmpTime(const Operation &a,const Operation &b){

    return a.t==b.t ? a.x<b.x : a.t<b.t;

}

int c[N];

inline int lowbit(int x){return x&-x;}

inline void add(int p,int v){for(;p<=n;p+=lowbit(p)) c[p]+=v;}

inline int sum(int p){

    int re=;

    for(;p;p-=lowbit(p)) re+=c[p];

    return re;

}

ll ans[N];

void CDQ(int l,int r){

    if(l==r) return;

    int mid=(l+r)>>;

    CDQ(l,mid);CDQ(mid+,r);

    int i=l,j=mid+,p=l;

    while(i<=mid||j<=r){

        if(j>r||(i<=mid&&a[i]<a[j])) add(a[i].y,),t[p++]=a[i++];

        else ans[a[j].t]+=sum(n)-sum(a[j].y),t[p++]=a[j++];

    }

    for(int i=l;i<=mid;i++) add(a[i].y,-);

    for(int i=l;i<=r;i++) a[i]=t[i];

    for(int i=r;i>=l;i--){

        if(a[i].t<=mid) add(a[i].y,);

        else ans[a[i].t]+=sum(a[i].y);

    }

    for(int i=l;i<=r;i++) if(a[i].t<=mid) add(a[i].y,-);

}

int main(){

    //freopen("inverse.in","r",stdin);

    //freopen("inverse.out","w",stdout);

    n=read();m=read();

    for(int i=;i<=n;i++) a[i]=Operation(,i,read()),mp[a[i].y]=i;

    int Tim=n;

    for(int i=;i<=m;i++) a[mp[read()]].t=Tim--;

    for(int i=;i<=n;i++) if(!a[i].t) a[i].t=Tim--;

    sort(a+,a++n,cmpTime);

    //for(int i=1;i<=10;i++) printf("hi %d %d %d\n",i,a[i].t,a[i].y);

    CDQ(,n);

    //for(int i=1;i<=10;i++) printf("ans %d %d\n",i,ans[i]);

    for(int i=;i<=n;i++) ans[i]+=ans[i-];

    for(int i=n;i>=n-m+;i--) printf("%lld\n",ans[i]);

}

【update 2017-03-17】

前两天学到了删除的姿势，逆序对问题的删除操作不用时间倒流也可以，直接减去它形成的逆序对数并且在树状数组中删除就可以了

虽然慢一些但是清晰多了

#include <iostream>

#include <cstdio>

#include <algorithm>

#include <cstring>

using namespace std;

typedef long long ll;

const int N=2e5+;

inline int read(){

    char c=getchar();int x=,f=;

    while(c<''||c>''){if(c=='-')f=-;c=getchar();}

    while(c>=''&&c<=''){x=x*+c-'';c=getchar();}

    return x*f;

}

int n,Q,a[N],pos[N],x;

int m,tim;

struct meow{

    int t,x,y,type,qid;

    meow(){}

    meow(int a,int b,int c,int d,int e=):t(a),x(b),y(c),type(d),qid(e){}

    bool operator <(const meow &r) const{

        return x==r.x ? y<r.y : x<r.x;

    }

}q[N],t[N];

ll ans[N];

int c[N];

inline void add(int p,int v) {for(;p<=n;p+=(p&-p)) c[p]+=v;}

inline int sum(int p) {int re=; for(;p;p-=(p&-p)) re+=c[p]; return re;}

void CDQ(int l,int r){

    if(l==r) return;

    int mid=(l+r)>>;

    for(int i=l;i<=r;i++){

        if(q[i].t<=mid) add(q[i].y,q[i].type);

        else ans[q[i].qid]+= q[i].type*( sum(n)-sum(q[i].y) );

    }

    for(int i=l;i<=r;i++) if(q[i].t<=mid) add(q[i].y,-q[i].type);

    for(int i=r;i>=l;i--){

        if(q[i].t<=mid) add(q[i].y,q[i].type);

        else ans[q[i].qid]+= q[i].type*sum(q[i].y-);

    }

    for(int i=l;i<=r;i++) if(q[i].t<=mid) add(q[i].y,-q[i].type);

    int p1=l,p2=mid+;

    for(int i=l;i<=r;i++){

        if(q[i].t<=mid) t[p1++]=q[i];

        else t[p2++]=q[i];

    }

    for(int i=l;i<=r;i++) q[i]=t[i];

    CDQ(l,mid); CDQ(mid+,r);

}

int main(){

    freopen("in","r",stdin);

    n=read(); Q=read();

    for(int i=;i<=n;i++) a[i]=read(), pos[a[i]]=i, q[++m]=meow(++tim, i, a[i], , );

    for(int i=;i<=Q;i++) x=read(), q[++m]=meow(++tim, pos[x], x, -, i);

    sort(q+, q++m);

    CDQ(,m);

    for(int i=;i<=Q;i++) ans[i]+=ans[i-],printf("%lld\n",ans[i-]);

}

巴特西

BZOJ 3295: [Cqoi2011]动态逆序对 [CDQ分治]

最新文章

热门文章