bzoj 2648 KD-tree

稍微看了一下KD-tree的讲义，大概明白了它的原理，但是实现不出来。。。

所以无耻的抄了一下黄学长的。。。

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #define ll long long
 #define inf 1000000000
 using namespace std;
 inline int read()
 {
     int x=,f=;char ch=getchar();
     while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
     while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
     return x*f;
 }
 int n,m,root,D;
 struct node
 {
     int d[],mn[],mx[],l,r;
     int& operator[](int x){return d[x];}
     node(int x=,int y=)
     {
         l=,r=,d[]=x,d[]=y;
     }
 }p[];
 bool operator <(node a,node b)
 {
     return a[D]<b[D];
 }
 inline int dis(node a,node b)
 {
     return abs(a[]-b[])+abs(a[]-b[]);
 }
 struct kdtree
 {
     int ans;
     node t[],T;
     void update(int k)
     {
         node l=t[t[k].l],r=t[t[k].r];
         for(int i=;i<;i++)
         {
             if(t[k].l)t[k].mn[i]=min(t[k].mn[i],l.mn[i]),t[k].mx[i]=max(t[k].mx[i],l.mx[i]);
             if(t[k].r)t[k].mn[i]=min(t[k].mn[i],r.mn[i]),t[k].mx[i]=max(t[k].mx[i],r.mx[i]);
         }
     }
     int build(int l,int r,int now)
     {
         D=now;
         int mid=(l+r)>>;
         nth_element(p+l,p+mid,p+r+);
         t[mid]=p[mid];
         for(int i=;i<;i++)
          t[mid].mn[i]=t[mid].mx[i]=t[mid][i];
         if(l<mid)t[mid].l=build(l,mid-,now^);
         if(r>mid)t[mid].r=build(mid+,r,now^);
         update(mid);
         return mid;
     }
     void insert(int k,int now)
     {
         if(T[now]>=t[k][now])
         {
             if(t[k].r)insert(t[k].r,now^);
             else
             {
                 t[k].r=++n;t[n]=T;
                 for(int i=;i<;i++)
                   t[n].mn[i]=t[n].mx[i]=t[n][i];
             }
         }
         else
         {
             if(t[k].l)insert(t[k].l,now^);
             else
             {
                 t[k].l=++n;t[n]=T;
                 for(int i=;i<;i++)
                     t[n].mn[i]=t[n].mx[i]=t[n][i];
             }
         }
         update(k);
     }
     int get(int k,node p)
     {
         int tmp=;
         for(int i=;i<;i++)
          tmp+=max(,t[k].mn[i]-p[i]);
         for(int i=;i<;i++)
          tmp+=max(,p[i]-t[k].mx[i]);
          return tmp;
     }
     void query(int k,int now)
     {
         int d,dl=inf,dr=inf;
         d=dis(t[k],T);
         ans=min(ans,d);
         if(t[k].l)dl=get(t[k].l,T);
         if(t[k].r)dr=get(t[k].r,T);
         if(dl<dr)
         {
             if(dl<ans)query(t[k].l,now^);
             if(dr<ans)query(t[k].r,now^);
         }
         else
         {
             if(dr<ans)query(t[k].r,now^);
             if(dl<ans)query(t[k].l,now^);
         }
     }
     int query(node p)
     {
         ans=inf;T=p;query(root,);
         return ans;
     }
     void insert(node p)
     {
         T=p;insert(root,);
     }
 }kd;
 int main()
 {
     n=read();m=read();
     for(int i=;i<=n;i++)p[i][]=read(),p[i][]=read();
     root=kd.build(,n,);
     while(m--)
     {
         int opt=read(),x=read(),y=read();
         if(opt==)kd.insert(node(x,y));
         else printf("%d\n",kd.query(node(x,y)));
     }
     return ;
 }