BZOJ 3262 陌上花开 - CDQ分治

最近又研究了一下cdq分治，发现比树套树好写多了啊。。。

首先cdq分治只能优化掉一维，对于一个三维问题，cdq分治先二分第一维，然后通过sort第二维后由左到右扫描确保第二维的顺序，第三维则需要一个计数的数据结构维护，一般是树状数组或者线段树，然后就完美地解决掉了问题。

而对于cdq递归时，可以采用先序遍历，然后再对子区间排序合并成大区间；也可后序遍历，在query完后拆成两个对于二维已排序的子区间。

以下是代码，注意注释部分。

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<algorithm>

using namespace std;

const int maxn=100005;

struct node
{
	int x,y,z,id,num,res;
	bool operator == (const node a) const
	{
		return (a.x==x)&&(a.y==y)&&(a.z==z);
	}
	bool operator < (const node a) const
	{
		if(a.y==y)
		{
			if(a.z==z)return x<a.x;
			return z<a.z;
		}
		return y<a.y;
	}
}q[maxn],point[maxn];

int n,m,cnt;
int c[maxn<<1];
int res[maxn],rank[maxn];

bool cmp(node a,node b)
{
	if(a.x==b.x)
	{
		if(a.y==b.y)return a.z<b.z;
		return a.y<b.y;
	}
	return a.x<b.x;
}
void update(int x,int val)
{
	for(int i=x;i<=m;i+=i&(-i))
		c[i]+=val;
}
int query(int x)
{
	int sum=0;
	for(int i=x;i;i-=i&(-i))
		sum+=c[i];
	return sum;
}
void cdq(int lt,int rt)
{
	if(lt==rt)
		return;
	int mid=lt+rt>>1;
	
	int p1=lt-1,p2=mid;
	for(int i=lt;i<=rt;i++) 
	{
		if(q[i].x<=mid)point[++p1]=q[i];
		else point[++p2]=q[i];
	}
	for(int i=lt;i<=rt;i++)
		q[i]=point[i];
		
	int last=lt;
	for(int i=mid+1;i<=rt;i++)
	{
		for(;last<=mid&&q[last].y<=q[i].y;last++)
			update(q[last].z,q[last].num);
		q[i].res+=query(q[i].z);
	}
	for(int i=lt;i<last;i++)update(q[i].z,-q[i].num);
	cdq(lt,mid);
	cdq(mid+1,rt);
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		scanf("%d%d%d",&point[i].x,&point[i].y,&point[i].z),point[i].id=i;
	//把x视为cdq分治的第一维，即时间
	sort(point+1,point+n+1,cmp);
	
	q[cnt=1]=point[1],q[1].num=1;
	
	for(int i=2;i<=n;i++)//处理全等的花，将其合并成一个区间 
	{
		if(!(point[i]==point[i-1]))
			q[++cnt]=point[i];
		q[cnt].num++;
	}
	for(int i=1;i<=cnt;i++)q[i].x=i;
	//注意这里a是可重复的，其最大值为m，而对于分治来说必须要有一个不重复的排列，所以做一个离散，使之最大值为cnt 
	sort(q+1,q+cnt+1);
	cdq(1,cnt);
	
	for(int i=1;i<=cnt;i++)
		rank[q[i].res+q[i].num-1]+=q[i].num;
	//注意一种的rank还需包括等于的，对于这种花的每一株还需加上num-1才为rank值 
	for(int i=0;i<n;i++)
		printf("%d\n",rank[i]);

	return 0;
}