模板题而已
不会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;
}