BZOJ 3262 陌上花开 三维偏序 树状数组套Treap

一共三维，第一维排序，第二维树状数组，第三维Treap

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
#include<cmath>
using namespace std;
const int maxn = 3000050;
int size,n,k,p,val,x,y,z,ans,sum[maxn],tmp[maxn];
int root[maxn];
struct node
{
    int x,y,z;
};
node a[maxn];
struct Treap
{
    int v,cnt,sz,ch[2],rnd;
}T[maxn];
bool cmp(node a,node b)
{
    if(a.x==b.x&&a.y==b.y) return a.z<b.z;
    if(a.x==b.x) return a.y<b.y;
    return a.x<b.x;
}
void Upd(int k){
    T[k].sz=T[T[k].ch[0]].sz+T[T[k].ch[1]].sz+T[k].cnt;
}
void rot(int &k,bool f){
    int t=T[k].ch[f];T[k].ch[f]=T[t].ch[!f];
    T[t].ch[!f]=k;Upd(k);Upd(t);k=t;
}
void insert(int &k){
    if(!k){k=++size;T[k].rnd=rand();T[k].v=val;T[k].cnt=T[k].sz=1;return;}
    T[k].sz++;
    if(T[k].v==val){T[k].cnt++;return;}
    bool f=val>T[k].v;insert(T[k].ch[f]);
    if(T[T[k].ch[f]].rnd<T[k].rnd)rot(k,f);
}
void rank(int k){
    if(!k)return;
    if(T[k].v==val)ans+=T[T[k].ch[0]].sz;
    else if(T[k].v>val)rank(T[k].ch[0]);
    else ans+=T[T[k].ch[0]].sz+T[k].cnt,rank(T[k].ch[1]);
}
int low(int x){
    return x&(-x);
}
void update(int x,int p){
    val=p;
    for(int i=x;i<=k;i+=low(i)) insert(root[i]); 
}
void query(int x,int p)
{
    val=p+1;
    for(int i=x;i;i-=low(i)) rank(root[i]);
}
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++){
        scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].z);
    } 
    sort(a+1,a+1+n,cmp);
    for(int i=1;i<=n;i++)
    {
        if(a[i].x==a[i+1].x&&a[i].y==a[i+1].y&&a[i].z==a[i+1].z) sum[i+1]=sum[i]+1;
        else
        {
            ans=0;
            query(a[i].y,a[i].z);
            tmp[ans]+=sum[i]+1; 
            //cout<<ans<<" "<<a[i].y<<" "<<a[i].z<<endl;
        } 
        update(a[i].y,a[i].z);
    }
    for(int i=0;i<n;i++)
        printf("%d\n",tmp[i]);
    return 0;
}