洛谷P3810 陌上花开 （cdq）

最近才学了cdq，所以用cdq写的代码（这道题也是cdq的模板题）

这道题是个三维偏序问题，先对第一维排序，然后去掉重复的，然后cdq分治即可。

为什么要去掉重复的呢？因为相同的元素互相之间都能贡献，而cdq过程中只能左边贡献右边的，所以要去重。

#include<bits/stdc++.h>
using namespace std;
const int N=200005;
struct node{
    int a,b,c,cnt,ans;
}s1[N],s2[N];
int n,k,mx,m,top,su[N];
int c[N];//树状数组 
 
bool cmp1(node x,node y){//按a排序 
    if(x.a==y.a){
        if(x.b==y.b) return x.c<y.c;
        else return x.b<y.b;
    }
    else return x.a<y.a;
}
 
bool cmp2(node x,node y){//cdq分治过程中对b排序 
    if(x.b==y.b) return x.c<y.c;
    else return x.b<y.b;
}
 
int lowbit(int x){
    return x&(-x);
}
 
void add(int x,int k){
    while(x<=mx){
        c[x]+=k;
        x+=lowbit(x);
    }
}
 
int query(int x){
    int sum=0;
    while(x){
        sum+=c[x];
        x-=lowbit(x);
    }
    return sum;
}
 
void cdq(int l,int r){//cdq 
    if(l==r) return ;
    int mid=(l+r)>>1;
    cdq(l,mid);cdq(mid+1,r);
    sort(s2+l,s2+1+mid,cmp2);
    sort(s2+mid+1,s2+r+1,cmp2);
    int i,j=l;
    for(int i=mid+1;i<=r;i++){//双指针计算结果 
        while(s2[i].b>=s2[j].b&&j<=mid){
            add(s2[j].c,s2[j].cnt);
            j++;
        }
        s2[i].ans+=query(s2[i].c);//计算ans 
    }
    for(int i=l;i<j;i++){//清空数组 
        add(s2[i].c,-s2[i].cnt);
    }
}
 
int main()
{
    scanf("%d%d",&n,&k);
    mx=k;
    for(int i=1;i<=n;i++){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        s1[i].a=a;s1[i].b=b;s1[i].c=c;
    }
    sort(s1+1,s1+1+n,cmp1); 
    for(int i=1;i<=n;i++){//去掉重复的 
        top++;
        if(s1[i].a!=s1[i+1].a||s1[i].b!=s1[i+1].b||s1[i].c!=s1[i+1].c){
            m++;
            s2[m].a=s1[i].a;s2[m].b=s1[i].b;s2[m].c=s1[i].c;
            s2[m].cnt=top;
            top=0;
        }
    }
    cdq(1,m);
    for(int i=1;i<=m;i++) su[s2[i].ans+s2[i].cnt-1]+=s2[i].cnt;
    for(int i=0;i<n;i++)
    cout<<su[i]<<endl; 
    return 0;
}