【HDU5992】Finding Hotels 【KD树】

题意

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

分析

  kd树的模板题

  

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