BZOJ 3262 陌上花开 CDQ分治

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=1000100;
struct node
{
int a,b,c,cnt,ans,id,op;
};
node T[maxn],P[maxn];
int f[maxn],n,m,ans[maxn],res[maxn];
int low(int x)
{
return x&(-x);
}
void add(int x,int val)
{
for(int i=x;i<=m;i+=low(i)) f[i]+=val;
}
int sum(int x)
{
int res=0;
for(int i=x;i;i-=low(i)) res+=f[i];
return res;
}
bool cmp1(node i,node j)
{
if(i.a==j.a&&i.b==j.b) return i.c<j.c;
if(i.a==j.a) return i.b<j.b;
return i.a<j.a;
}

bool cmpCDQ(node i,node j)
{
if(i.b!=j.b) return i.b<j.b;
return i.op<j.op;
}

void solve(int l,int r)//solve(l,r)为求出按照第一维排序后的区间[l,r]对答案的贡献
{
if(l==r)return ;
int mid=(l+r)>>1;
solve(l,mid);solve(mid+1,r);//保证调用完两个子过程时，区间[l,mid]和区间[mid+1,r]中第二维是有序的.
int cnt=0;
for(int i=l;i<=mid;i++) P[++cnt]=T[i],P[cnt].op=1;
for(int i=mid+1;i<=r;i++) P[++cnt]=T[i],P[cnt].op=2;
sort(P+1,P+cnt+1,cmpCDQ);
for(int i=1;i<=cnt;i++)
{
if(P[i].op==1) add(P[i].c,P[i].cnt);
else ans[P[i].id]+=sum(P[i].c);
}
for(int i=1;i<=cnt;i++) if(P[i].op==1) add(P[i].c,-P[i].cnt);
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d%d%d",&T[i].a,&T[i].b,&T[i].c);
T[i].cnt=1;
}
sort(T+1,T+n+1,cmp1);
int tot=1;
for(int i=2;i<=n;i++)
{
if(T[i].a!=T[tot].a||T[i].b!=T[tot].b||T[i].c!=T[tot].c)
{
tot++;
T[tot]=T[i];
}
else T[tot].cnt++;
}
for(int i=1;i<=tot;i++) T[i].id=i;
solve(1,tot);
for(int i=1;i<=tot;i++)
{
res[ans[T[i].id]+T[i].cnt-1]+=T[i].cnt;
}
for(int i=0;i<n;i++) printf("%d\n",res[i]);
return 0;
}