P3810 【模板】三维偏序(陌上花开)

模板题而已

不会cdq的看这里:cdq分治:从分治到套娃

一维排序,二维cdq,三维树状数组

#include<algorithm>
#include<iostream>
#include<cstdio>
using namespace std;
int n1,n2,k;
const int N=100010,K=200010;
int ans[N],tr[K];
struct treAKer{int a,b,c,num,ans;}a[N],b[N];
int my_a(const treAKer &x,const treAKer &y)//自定义排序 
{//以 a第一关键字 b第二关键字 c第三关键字 排序 
    if(x.a==y.a)
    {
        if(x.b==y.b)return x.c<y.c;
        return x.b<y.b;
    }
    return x.a<y.a;//满足x.a<y.a这个条件的话就让x排在前边 
}
int my_b(const treAKer &x,const treAKer &y)//重点:满足return后面的条件的话就让x排在前边
{//以 b第一关键字 c第二关键字 排序 
    if(x.b==y.b)return x.c<y.c; 
    return x.b<y.b;
}

int lowbit(int x){return x&(-x);}
int ask(int x)
{
	int ans=0;
	for(;x;x-=lowbit(x))ans+=tr[x];
	return ans;
}
void change(int x,int v){for(;x<=k;x+=lowbit(x))tr[x]+=v;}

void cdq(int l,int r)//重点:分治 
{
	if(l==r)return;
	int mid=(l+r)>>1;
	cdq(l,mid);cdq(mid+1,r);
	//到上边为止都是基础操作 上边是分治过程中的分解问题,下面开始合并 
	sort(b+l,b+mid+1,my_b);sort(b+mid+1,b+r+1,my_b);//排完序后左右的b都是从小到大,a是乱序的,但是,即使a是乱序的,右边所有的a都比左边所有的a大(这是cdq分治的重点) 
	//到现在位置,左右两块块内b有序,并且右边的a大于左边的a,此时a不用考虑,b可以用双指针 
	int i=l,j=mid+1;
	for(;j<=r;++j)
	{
		while(i<=mid&&b[i].b<=b[j].b/*while用来保证第二维有序*/)change(b[i].c,b[i].num),++i;
		b[j].ans+=ask(b[j].c);//用树状数组维护第三维的信息 
	}
	for(int k=l;k<=i-1;++k)change(b[k].c,-b[k].num);//记得把树状数组还原 
}
int main()
{
	cin>>n1>>k;
	for(int i=1;i<=n1;++i)scanf("%d%d%d",&a[i].a,&a[i].b,&a[i].c);
	sort(a+1,a+n1+1,my_a);
	for(int i=1,now=0;i<=n1;++i)//去重 
	{
		++now;//now记录重复的有多少个 
		if(a[i].a!=a[i+1].a||a[i].b!=a[i+1].b||a[i].c!=a[i+1].c)b[++n2]=a[i],b[n2].num=now,now=0/*去重操作 把多个相同的合并成一个 num记录重复的个数*/;
	}
	cdq(1,n2);
    for(int i=1;i<=n2;++i)ans[b[i].ans+b[i].num-1]+=b[i].num;//统计答案 
	for(int i=0;i<n1;++i)printf("%d\n",ans[i]);
	return 0;
}