[BZOJ 3262陌上花开] CDQ分治

[BZOJ 3262]

• 一个区间的答案是左右两个子区间单独的答案加左右两个子区间之间产生的新贡献，所以就可以分治了
• 先按第一维排序
• 分别在每个分治的子区间里按第二维排序，类似归并排序
• 第三维用树状数组维护即可

Code

---

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
using namespace std;
const int N=4e5;
struct node{int x,y,z,rk,cnt;}a[N];
int n,k,tot=0,c[N],ans[N];
int lowbit(int x){return x&-x;}
void add(int x,int v){while(x<=k)c[x]+=v,x+=lowbit(x);}
int query(int x){int sum=0;while(x)sum+=c[x],x-=lowbit(x);return sum;}
int read(){
	int num=0;char ch=getchar();
	while(isdigit(ch)==0)ch=getchar();
	while(isdigit(ch)!=0){num=num*10+ch-'0';ch=getchar();}
	return num;
}
bool cmp1(node x,node y)
{
	return x.x<y.x||x.x==y.x&&x.y<y.y||x.x==y.x&&x.y==y.y&&x.z<y.z;
}
bool cmp2(node x,node y)
{
	return x.y<y.y||x.y==y.y&&x.z<y.z;
}
void cdq(int l,int r){
	if(l==r){
		a[l].rk+=a[l].cnt-1;
		return;
	}
	int mid=l+r>>1;
	cdq(l,mid),cdq(mid+1,r);
	sort(a+l,a+mid+1,cmp2);
	sort(a+mid+1,a+r+1,cmp2);
	int j=l;
	rep(i,mid+1,r){
		while(j<=mid&&a[j].y<=a[i].y)
		add(a[j].z,a[j].cnt),j++;
		a[i].rk+=query(a[i].z);
	}
	rep(i,l,j-1)add(a[i].z,-a[i].cnt);
}
int main()
{
	n=read(),k=read();
	rep(i,1,n)a[i].x=read(),a[i].y=read(),a[i].z=read(),a[i].rk=1;
	sort(a+1,a+n+1,cmp1);
	rep(i,1,n){
		if(i!=1&&a[i].x==a[i-1].x&&a[i].y==a[i-1].y&&a[i].z==a[i-1].z)
		a[tot].cnt++;
		else a[++tot]=a[i],a[tot].cnt=1;
	}
	cdq(1,tot);
	rep(i,1,tot)ans[a[i].rk]+=a[i].cnt;
	rep(i,1,n)printf("%d\n",ans[i]);
	return 0;
}