【HDU5992】Finding Hotels 【KD树】

题意

给出n个酒店的坐标和价格，然后m个查询，每个查询给出一个人的坐标和能承受的最大价格，然后找出在他价格承受范围以内，距离他最近的宾馆，如果有多个，那么输出第一个

分析

kd树的模板题

     #include <cstdio>
     #include <cstring>
     #include <algorithm>
     #include <iostream>
     #include <cmath>
 
     using namespace std;
     const int maxn=+;
     const int INF=;
 
     typedef long long LL;
     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;
     }
     LL ANS;
     struct kdNode{
         int x[];
         int div;
         int id;
         bool lef;
     }ans,p[maxn],q;
     int cmpNo;
     bool cmp(kdNode a,kdNode b){
         return a.x[cmpNo]<b.x[cmpNo];
     }
 
     inline LL dis(kdNode a,kdNode b){
         LL res=(LL)(a.x[]-b.x[])*(a.x[]-b.x[])+(LL)(a.x[]-b.x[])*(a.x[]-b.x[]);
         return res;
     }
     void buildKD(int l,int r,kdNode* p,int d){
         if(l>r)return;
         int m=(l+r)/;
         cmpNo=d;
         nth_element(p+l,p+m,p+r+,cmp);
         p[m].div=d;
      //    printf("%d %d %lld %lld %lld\n",l,r,p[m].x[0],p[m].x[1],p[m].x[2]);
         if(l==r){
             p[m].lef=;
             return;
         }
         buildKD(l,m-,p,(d+)%);
         buildKD(m+,r,p,(d+)%);
     }
     void findkd(int l,int r,kdNode &tar,kdNode *p){
         if(l>r)return ;
         int m=(l+r)/;
         LL d=dis(p[m],tar);
         if(p[m].lef){
             if(p[m].x[]<=tar.x[]){
                 if(ANS>d||(ANS==d&&ans.id>p[m].id)){
                     ANS=d;
                     ans.id=p[m].id;
                     for(int i=;i<=;i++)
                         ans.x[i]=p[m].x[i];
                 }
             }
             return;
         }
         int t=tar.x[p[m].div]-p[m].x[p[m].div];
         if(t>){
             findkd(m+,r,tar,p);
             if(p[m].x[]<=tar.x[]){
                 if(ANS>d||(ANS==d&&p[m].id<ans.id)){
                     ANS=d;
                     for(int j=;j<=;j++)
                         ans.x[j]=p[m].x[j];
                     ans.id=p[m].id;
                 }
             }
             if(ANS>t)
                 findkd(l,m-,tar,p);
         }else{
             findkd(l,m-,tar,p);
             if(p[m].x[]<=tar.x[]){
                 if(ANS>d||(ANS==d&&p[m].id<ans.id)){
                     ANS=d;
                     for(int j=;j<=;j++)
                         ans.x[j]=p[m].x[j];
                     ans.id=p[m].id;
                 }
             }
             if(ANS>-t)
                 findkd(m+,r,tar,p);
         }
     }
 
     int T,n,m;
     int main(){
         T=read();
         for(int kas=;kas<=T;kas++){
           //  scanf("%d%d",&n,&m);
             n=read(),m=read();
             for(int i=;i<n;i++){
                 //scanf("%d%d%d",&p[i].x[0],&p[i].x[1],&p[i].x[2]);
                 p[i].x[]=read();p[i].x[]=read();p[i].x[]=read();
                 p[i].lef=;
                 p[i].id=i;
               //  printf("%lld %lld %lld ",p[i].x[0],p[i].x[1],p[i].x[2]);
             }
             buildKD(,n-,p,);
             for(int i=;i<=m;i++){
                 ANS=INF;
                 //scanf("%d%d%d",&q.x[0],&q.x[1],&q.x[2]);
                 q.x[]=read();q.x[]=read();q.x[]=read();
                 findkd(,n-,q,p);
                 printf("%d %d %d\n",ans.x[],ans.x[],ans.x[]);
             }
         }
     return ;
     }