BZOJ 3262 陌上花开 CDQ分治

题目大致就是求一个三维偏序，我们可以排序第一维，第二维用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;
}