P3810 【模板】三维偏序（陌上花开）

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=1e7+10;
int ans[maxn];
int f[maxn];
int n,k;
int c[maxn];
int same[maxn];
struct node
{
    int  a,b,c;
    int  w;
}way[maxn];
struct XX
{
    
int lowbit(int  x)
{
    return x&(-x);
}
void add(int  x,int  w)
{
    for(;x<=k;x+=lowbit(x)) 
    {
        c[x]+=w;
    }
}
int sum(int  x)
{
    int  res=0;
    for(;x;x-=lowbit(x))
    {
        res+=c[x];
    }
    return res;
}
}XX;
bool cmp1(node a,node b)
{
    if(a.a!=b.a)
    {
        return a.a<b.a;
    }
    if(a.b!=b.b)
    {
        return a.b<b.b;
    }
    return a.c<b.c;
}

bool cmp2(node a,node b)
{
    if(a.b!=b.b)
    {
        return a.b<b.b;
    }
    if(a.c!=b.c)
    {
        return a.c<b.c;
    }
    return a.a<b.a;
}
void  CDQ(int  l,int  r)
{
    if(l==r)
    {
        return ;
    }
    int  mid=(l+r)>>1;
    CDQ(l,mid);
    CDQ(mid+1,r);
    sort(way+l,way+1+r,cmp2);
    for(int i=l;i<=r;i++)
    {
        if(way[i].a<=mid)
        {
            XX.add(way[i].c,1);
        }
        else
        {
            ans[way[i].w]+=XX.sum(way[i].c);
        }
    }
    for(int i=l;i<=r;i++)
    {
        if(way[i].a<=mid)
        {
            XX.add(way[i].c,-1);
        }
    }
}
    
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d%d",&way[i].a,&way[i].b,&way[i].c);
        way[i].w=i;
    }
    sort(way+1,way+1+n,cmp1);
    int i=1;
    while(i<=n)
    {
        int j=i+1;
        while(j<=n&&way[i].a==way[j].a&&way[i].b==way[j].b&&way[i].c==way[j].c)
        {
            j++;
        }
        while(i<j)
        {
            same[way[i++].w]=way[j-1].w;
        }
    }
    for(int i=1;i<=n;i++)
    {
        way[i].a=i;
    }
    CDQ(1,n);
    for(int i=1;i<=n;i++)
    {
        f[ans[same[way[i].w]]]++;
    }
    for(int i=0;i<n;i++)
    {
        printf("%d
",f[i]);
    }
    return 0;
}