BZOJ4066 简单题（KD-Tree）

　　板子题。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cassert>
using namespace std;
#define ll long long
#define N 200010
#define inf 2000000000
#define lson tree[k].ch[0]
#define rson tree[k].ch[1]
char getc(){char c=getchar();while ((c<'A'||c>'Z')&&(c<'a'||c>'z')&&(c<''||c>'')) c=getchar();return c;}
int gcd(int n,int m){return m==?n:gcd(m,n%m);}
int read()
{
    int x=,f=;char c=getchar();
    while (c<''||c>'') {if (c=='-') f=-;c=getchar();}
    while (c>=''&&c<='') x=(x<<)+(x<<)+(c^),c=getchar();
    return x*f;
}
int n,m,root,cnt,c,ans,t,tot,id[N];
const double alpha=0.75;
struct point
{
    int d[],v;
    bool operator <(const point&a) const
    {
        return d[c]<a.d[c];
    }
}a[N];
struct KDTree{int ch[],a[][],size,sum;point p;
}tree[N];
inline bool isin(point p,int a[][]){return p.d[]>=a[][]&&p.d[]<=a[][]&&p.d[]>=a[][]&&p.d[]<=a[][];}
inline bool isin(int a[][],int b[][]){return a[][]>=b[][]&&a[][]<=b[][]&&a[][]>=b[][]&&a[][]<=b[][];}
inline bool iscross(int a[][],int b[][]){return max(a[][],b[][])<=min(a[][],b[][])&&max(a[][],b[][])<=min(a[][],b[][]);}
void newnode(int k,point p)
{
    tree[k].size=,tree[k].sum=p.v,tree[k].p=p;
    tree[k].a[][]=tree[k].a[][]=p.d[],tree[k].a[][]=tree[k].a[][]=p.d[];
    lson=rson=;
}
void get(int k)
{
    if (lson) get(lson);
    t++,id[t]=k,a[t]=tree[k].p;
    if (rson) get(rson);
}
void build(int &k,int l,int r,int op)
{
    if (l>r) return;
    c=op;
    int mid=l+r>>;
    nth_element(a+l,a+mid,a+r+);
    newnode(k=id[++tot],a[mid]);
    for (int i=l;i<=r;i++)
    tree[k].a[][]=min(tree[k].a[][],a[i].d[]),tree[k].a[][]=max(tree[k].a[][],a[i].d[]),
    tree[k].a[][]=min(tree[k].a[][],a[i].d[]),tree[k].a[][]=max(tree[k].a[][],a[i].d[]);
    build(lson,l,mid-,op^),build(rson,mid+,r,op^);
    tree[k].size+=tree[lson].size,tree[k].size+=tree[rson].size,
    tree[k].sum+=tree[lson].sum,tree[k].sum+=tree[rson].sum;
}
void rebuild(int &k,int op)
{
    t=;get(k);
    tot=;build(k,,t,op);
}
void ins(int &k,point p,int op)
{
    if (!k) {newnode(k=++cnt,p);return;}
    if (max(tree[lson].size,tree[rson].size)>tree[k].size*alpha) rebuild(k,op);
    tree[k].size++;tree[k].sum+=p.v;
    tree[k].a[][]=min(tree[k].a[][],p.d[]),tree[k].a[][]=max(tree[k].a[][],p.d[]);
    tree[k].a[][]=min(tree[k].a[][],p.d[]),tree[k].a[][]=max(tree[k].a[][],p.d[]);
    if (tree[k].a[op][]<=p.d[]) ins(lson,p,op^);
    else ins(rson,p,op^);
}
int query(int k,int a[][])
{
    if (!k) return ;
    if (isin(tree[k].a,a)) return tree[k].sum;
    return (isin(tree[k].p,a)?tree[k].p.v:)+
    (iscross(tree[lson].a,a)?query(lson,a):)+
    (iscross(tree[rson].a,a)?query(rson,a):);
}
int main()
{
#ifndef ONLINE_JUDGE
    freopen("bzoj4066.in","r",stdin);
    freopen("bzoj4066.out","w",stdout);
    const char LL[]="%I64d\n";
#else
    const char LL[]="%lld\n";
#endif
    read();int op=read();
    while (op<)
    {
        if (op==)
        {
            int x=read()^ans,y=read()^ans,z=read()^ans;
            ins(root,(point){{x,y},z},);
        }
        else
        {
            int a[][];a[][]=read()^ans,a[][]=read()^ans,a[][]=read()^ans,a[][]=read()^ans;
            printf("%d\n",ans=query(root,a));
        }
        op=read();
    }
    return ;
}