cdq分治——P3157 [CQOI2011]动态逆序对

https://daniu.luogu.org/problem/show?pid=3157
本来想不看题解做的,结果差点调不出来;
标准的三维偏序;

对于一个三元组(a,b,c),有两组满足“逆序对”的条件
a2 < a && b 2 < b && c2 > c
a2 < a && b 2 > b && c2 < c

然后就是最后的时候要求一个前缀和;
至于为什么这个其实也很显然,但是我一开始的时候没有想到;

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define Ll long long
using namespace std;
const int N=100005;
struct cs{int i,v,num;}a[N],c[N];
Ll ans[N];
int f[N];
int n,m,top,x;
bool cmp(cs x,cs y){return x.num<y.num;}
bool cmp1(cs x,cs y){return x.i>y.i;}
bool cmp2(cs x,cs y){return x.i<y.i;}
void init(int i){for(;i<=n+1;i+=i&-i)f[i]++;}
void cle(int i){for(;i<=n+1;i+=i&-i)f[i]=0;}
int outit(int i){int ans=0;for(;i;i-=i&-i)ans+=f[i];return ans;}
void cdq(int l,int r){
    if(l==r)return;
    int mid=l+r>>1,k;
    cdq(l,mid);
    for(int i=l;i<=r;i++)c[i]=a[i];
    sort(c+l,c+mid+1,cmp1);
    sort(c+mid+1,c+r+1,cmp1);
    k=l-1;
    for(int i=mid+1;i<=r;i++){
        while(k<mid&&c[k+1].i>c[i].i)init(c[++k].v);
        ans[c[i].num]+=outit(c[i].v);
    }
    for(int i=l;i<=mid;i++)cle(c[i].v);
    sort(c+l,c+mid+1,cmp2);
    sort(c+mid+1,c+r+1,cmp2);
    k=l-1;
    for(int i=mid+1;i<=r;i++){
        while(k<mid&&c[k+1].i<c[i].i)init(n+1-c[++k].v);
        ans[c[i].num]+=outit(n+1-c[i].v);
    }
    for(int i=l;i<=mid;i++)cle(n+1-c[i].v);
    cdq(mid+1,r);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){scanf("%d",&x);a[x].v=x;a[x].i=i;}
    top=n;
    for(int i=1;i<=m;i++){scanf("%d",&x);a[x].num=top--;}
    for(int i=1;i<=n;i++)if(!a[i].num)a[i].num=top--;
    sort(a+1,a+n+1,cmp);
    cdq(1,n);
    for(int i=1;i<=n;i++)ans[i]+=ans[i-1];
    for(int i=0;i<m;i++)printf("%lld\n",ans[n-i]);
} 
posted @ 2017-04-24 13:41  largecube233  阅读(112)  评论(0编辑  收藏  举报