BZOJ 3262: 陌上花开

经典CDQ分治题，求三维偏序

#include<cstdio>
#include<algorithm>
using namespace std;

const int maxn=200000+200;

struct Node{

    int a,b,c;
    int num,level;
}node[maxn];

bool cmp(Node A,Node B){

    if(A.a==B.a&&A.b==B.b) return A.c<B.c;
    if(A.a==B.a) return A.b<B.b;
    return A.a<B.a; 
}

bool cmpcdq(Node A,Node B){

    if(A.b==B.b) return A.c<B.c;
    return A.b<B.b;
}

int tree[maxn],color[maxn];
int ans[maxn];
int type;
int n,k;
int tot;

inline int lowbit(int x){

    return x&(-x);
} 

void Add(int ind,int v){

    while(ind<=k){

        if(color[ind]!=type) tree[ind]=0;
        color[ind]=type;
        tree[ind]+=v;
        ind+=lowbit(ind);
    }
}

int getSum(int ind){

    int sum=0;
    while(ind>0){

        if(color[ind]==type) sum+=tree[ind];
        ind-=lowbit(ind);
    }
    return sum;
}


void CDQ(int l,int r){

    if(l==r) return ;
    int mid=(l+r)>>1;
    CDQ(l,mid);
    CDQ(mid+1,r);
    sort(node+l,node+mid+1,cmpcdq);
    sort(node+mid+1,node+r+1,cmpcdq);
    int i=l;
    int j=mid+1;
    type++;
    for(;j<=r;j++){

        for(;node[i].b<=node[j].b&&i<=mid;i++) Add(node[i].c,node[i].num);
        node[j].level+=getSum(node[j].c); 
    }
}

int main(){

    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++){

        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        node[i]=Node{a,b,c,1,0};
    }
    sort(node+1,node+n+1,cmp);
    tot=1;
    for(int i=2;i<=n;i++){

        if(node[i].a==node[tot].a&&node[i].b==node[tot].b&&node[i].c==node[tot].c) node[tot].num++;
        else node[++tot]=node[i];
    }
    CDQ(1,tot);
    for(int i=1;i<=tot;i++) ans[node[i].num+node[i].level-1]+=node[i].num;
    for(int i=0;i<n;i++) printf("%d\n",ans[i]);
}