洛谷P3810 陌上花开 （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;
}