【HDU5126】 stars k-d树

题目大意：有$m$个操作，分两种：在指定三维坐标内加入一个点，询问指定空间内点的数量。

其中$m≤5*10^{4},1≤x,y,z≤10^9$

这题几乎就是裸的$k-d$树啊。我们动态维护一棵$k-d$树，对于以某个节点为根的字树中，维护该子树中点的三维范围，以及该子树中点的数量。然后直接查询即可。

PS：为了保证你的程序不会变成“优秀”的$n^{2}$做法，需要写一个类似虚树暴力重构的东西，去控制该树的平衡。

#include<bits/stdc++.h>
#define M 100000
using namespace std;
struct node{
    int a[],max[],min[],sum,l,r,d;
    node(){
        a[]=a[]=a[]=; max[]=max[]=max[]=;
        min[]=min[]=min[]=; sum=d=l=r=;
    }
    node(int x,int y,int z){
        a[]=x; a[]=y; a[]=z; sum=l=d=r=;
        max[]=max[]=max[]=; min[]=min[]=min[]=;
    }
    node(int x1,int y1,int z1,int x2,int y2,int z2){
        max[]=x2; max[]=y2; max[]=z2;
        min[]=x1; min[]=y1; min[]=z1;
        a[]=a[]=a[]=sum=l=r=d=;
    }
}a[M]; int root=,use=;
int id[M]={},rebid=,rebfa=,now=;
void add(int &x,int fa,int d,node k){
    if(!x) x=++use,a[x]=k,a[x].d=d;
    else{
        if(k.a[d]<a[x].a[d]) add(a[x].l,x,(d+)%,k);
        else add(a[x].r,x,(d+)%,k);
    }
    a[x].max[]=max(a[x].max[],k.a[]); a[x].min[]=min(a[x].min[],k.a[]);
    a[x].max[]=max(a[x].max[],k.a[]); a[x].min[]=min(a[x].min[],k.a[]);
    a[x].max[]=max(a[x].max[],k.a[]); a[x].min[]=min(a[x].min[],k.a[]);
    a[x].sum++;
    if(max(a[a[x].l].sum,a[a[x].r].sum)>a[x].sum*0.7) rebid=x,rebfa=fa;
}
int D;
bool cmp(int x,int y){
    return a[x].a[D]<a[y].a[D];
}
void bl(int x){
    if(!x) return; id[++now]=x;
    bl(a[x].l); bl(a[x].r);
    a[x].max[]=a[x].max[]=a[x].max[]=a[x].l=a[x].r=a[x].sum=a[x].d=;
    a[x].min[]=a[x].min[]=a[x].min[]=;
}
void rebuild(int &x,int l,int r,int d){
    if(l>r) return; int mid=(l+r)>>; D=d;
    nth_element(id+l,id+mid,id+r+,cmp); x=id[mid];
    rebuild(a[x].l,l,mid-,(d+)%); rebuild(a[x].r,mid+,r,(d+)%);
    for(int i=;i<;i++)
    a[x].max[i]=max(a[x].a[i],max(a[a[x].l].max[i],a[a[x].r].max[i])),
    a[x].min[i]=min(a[x].a[i],min(a[a[x].l].min[i],a[a[x].r].min[i]));
    a[x].sum=a[a[x].l].sum+a[a[x].r].sum+; a[x].d=d;
}
void rebuild(){
    if(!rebid) return;
    bl(rebid);
    if(rebfa==) {rebuild(root,,now,a[rebid].d);}
    else{
        if(a[rebfa].l==rebid) rebuild(a[rebfa].l,,now,a[rebid].d);
        else rebuild(a[rebfa].r,,now,a[rebid].d);
    }
    rebid=rebfa=now=;
}
int query(int x,node k){
    int ck=,sum=; if(!x) return ;
    for(int i=;i<;i++) if(k.min[i]<=a[x].min[i]&&a[x].max[i]<=k.max[i]) ck++; if(ck==) return a[x].sum;
    for(int i=;i<;i++) if(k.max[i]<a[x].min[i]||a[x].max[i]<k.min[i]) return ; ck=;
    for(int i=;i<;i++) if(k.min[i]<=a[x].a[i]&&a[x].a[i]<=k.max[i]) ck++; if(ck==) sum++;
    return sum+query(a[x].l,k)+query(a[x].r,k);
}
int main(){
    int n; scanf("%d",&n);
    while(n--){
        int op,x,y,z,x1,y1,z1; scanf("%d%d%d%d",&op,&x,&y,&z);
        if(op==) add(root,,,node(x,y,z)),rebuild();
        else scanf("%d%d%d",&x1,&y1,&z1),printf("%d\n",query(root,node(x,y,z,x1,y1,z1)));
    }
}