【BZOJ3262】陌上花开（CDQ分治）

【BZOJ3262】陌上花开（CDQ分治）

题解

原来放过这道题目，题面在这里

树套树的做法也请点上面

这回用CDQ分治做的

其实也很简单，

对于第一维排序之后

显然只有前面的对后面的才会产生贡献

那么，使用CDQ分治

先分，每次递归子问题

合并的时候每次考虑前面的对于后面的贡献

最后统计一下答案

如果在清空树状数组的时候用了memset会TLE

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<queue>
#include<vector>
using namespace std;
#define MAX 110000
inline int read()
{
	int x=0,t=1;char ch=getchar();
	while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
	if(ch=='-')t=-1,ch=getchar();
	while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
	return x*t;
}
int c[MAX<<1],n,K;
int ans[MAX];
struct TNode{int a,b,c,w,ans;}t[MAX];
bool operator<(TNode x,TNode y)
{
	if(x.a!=y.a)return x.a<y.a;
	if(x.b!=y.b)return x.b<y.b;
	return x.c<y.c;
}
bool operator==(TNode a,TNode b)
{
	return a.a==b.a&&a.b==b.b&&a.c==b.c;
}
bool cmp(TNode x,TNode y)
{
	if(x.b!=y.b)return x.b<y.b;
	return x.c<y.c;
}
inline int lowbit(int x){return x&(-x);}
inline void Add(int x,int w){while(x<=K)c[x]+=w,x+=lowbit(x);}
inline int Getsum(int x){int ret=0;while(x)ret+=c[x],x-=lowbit(x);return ret;}
void CDQ(int l,int r)
{
	if(l==r)return;
	int mid=(l+r)>>1;
	CDQ(l,mid);CDQ(mid+1,r);
	sort(&t[l],&t[mid+1],cmp);
	sort(&t[mid+1],&t[r+1],cmp);
	int i=l;
	for(int j=mid+1;j<=r;++j)
	{
		while(i<=mid&&t[j].b>=t[i].b)
		{
			Add(t[i].c,t[i].w);
			++i;
		}
		t[j].ans+=Getsum(t[j].c);
	}
	for(int j=l;j<i;++j)Add(t[j].c,-t[j].w);
	//memset(c,0,sizeof(c));
}
int main()
{
	n=read();K=read();
	for(int i=1;i<=n;++i)
		t[i].a=read(),t[i].b=read(),t[i].c=read();
	sort(&t[1],&t[n+1]);
	int tot=0;
	for(int i=1;i<=n;++i)
	{
		if(t[i]==t[i-1])++t[tot].w;
		else t[++tot]=t[i],t[tot].w++;
	}
	int N=n;
	n=tot;
	CDQ(1,n);
	for(int i=1;i<=n;++i)ans[t[i].ans+t[i].w-1]+=t[i].w;
	for(int i=0;i<N;++i)
		printf("%d\n",ans[i]);
	return 0;
}