【BZOJ2716】天使玩偶【kd树】

这个题要求kd树支持两个操作。

1.插入一个新的点。

2.查询某个点最近曼哈顿距离。

注意查询曼哈顿距离和查询欧几里得距离，是有区别的。（估价函数不同）。

 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #include <iostream>
 #include <cmath>
 using namespace std;
 const int maxn=+;
 const int INF=;

 struct kdNode{
     int x[],mnn[],mxn[];
     int div;
     bool lef;
 }p[maxn],q;
 int n,m,cmpNo;
 int lc[maxn],rc[maxn];
 int cmp(kdNode a,kdNode b){
     return a.x[cmpNo]<b.x[cmpNo];
 }
 void maintain(int o){
     int ls=lc[o],rs=rc[o];
     for(int i=;i<;i++){
         p[o].mnn[i]=min(min(p[ls].mnn[i],p[rs].mnn[i]),p[o].x[i]);
         p[o].mxn[i]=max(max(p[ls].mxn[i],p[rs].mxn[i]),p[o].x[i]);
     }
 }

 void build(int&o,int l,int r,int d){
     if(l>r){
         o=;
         return;
     }
     int m=l+(r-l)/;
     cmpNo=d;
     nth_element(p+l,p+m,p+r+,cmp);
     o=m;
     p[m].div=d;
     if(l==r){//好像可以不写
         p[m].lef=;
         p[m].mnn[]=p[m].mxn[]=p[m].x[];
         p[m].mnn[]=p[m].mxn[]=p[m].x[];
         return;
     }
     build(lc[o],l,m-,d^);
     build(rc[o],m+,r,d^);
     maintain(o);
 }
 int cal(int o){
     int res=;
     for(int i=;i<;i++)
         res+=max(,p[o].mnn[i]-q.x[i]);
     for(int i=;i<;i++)
         res+=max(,q.x[i]-p[o].mxn[i]);
 }
 void Insert(int& o,int d){
     if(!o){
         o=++n;
         p[n]=q;
         p[n].div=d;
         p[n].lef=;
 //        p[n].mnn[0]=p[n].mxn[0]=p[n].x[0];
 //        p[n].mnn[1]=p[n].mxn[1]=p[n].x[1];
         return maintain(o);
         //return;
     }
     int t=q.x[d]-p[o].x[d];
     if(t>=){
         Insert(rc[o],d^);
     }else{
         Insert(lc[o],d^);
     }
     maintain(o);
 }
 int ans;
 void query(int o,int d){
     if(!o)return;
     ans=min(ans,abs(p[o].x[]-q.x[])+abs(p[o].x[]-q.x[]));
     int d1=cal(lc[o]),d2=cal(rc[o]);
     if(d2<d1){
         query(rc[o],d^);
         if(ans>d1)
             query(lc[o],d^);
     }else{
         query(lc[o],d^);
         if(ans>d2)
             query(rc[o],d^);
     }
 }

 int main(){
     freopen("in.txt","r",stdin);
     freopen("out.txt","w",stdout);
     scanf("%d%d",&n,&m);
     for(int i=;i<=n;i++){
         scanf("%d%d",&p[i].x[],&p[i].x[]);
     }
     int root;
     p[].mnn[]=p[].mnn[]=INF;
     p[].mxn[]=p[].mxn[]=-INF;
     build(root,,n,);
     for(int i=;i<=m;i++){
         int opt;
         scanf("%d",&opt);
         if(opt==){
             scanf("%d%d",&q.x[],&q.x[]);
             Insert(root,);
         }else{
             scanf("%d%d",&q.x[],&q.x[]);
             ans=INF;
             query(root,);
             printf("%d\n",ans);
         }
     }
 return ;
 }