Hdu-5992 2016ACM/ICPC亚洲区青岛站 K.Finding Hotels KDtree

题面

题意:二维平面上有很多点,每个点有个权值,现在给你一个点(很多组),权值v,让你找到权值小于等于v的点中离这个点最近的,相同的输出id小的

题解:很裸的KDtree,但是查询的时候有2个小限制,

1个是要小于等于v,1个是输出最小id

第一个,对每个点判断dis的时候 如果价钱高于v 距离就变为INF 低于v就没有影响

第二个,如果disl或者disr 和当前最近的dis 相等 就继续询问以得到更小的id

 #include<bits/stdc++.h>
 typedef long long ll;
 using namespace std;
 const int N=2e5+;
 const ll INF=1e18;
 int T,n,m;
 struct point
 {
     ll x[];
     int c,id;
 }p[N],d;
 ll idd,ansid,ans;
 ll sqr(ll x) {return x*x;}
 int cur;
 bool cmp(point a,point b)
 {
     return a.x[cur]<b.x[cur];
 }
 void build(int l,int r,int dep)
 {
     if(l>=r) return;
     int mid=(l+r)>>;
     cur=dep%;
     nth_element(p+l,p+mid,p+r+,cmp);
     build(l,mid-,dep+);
     build(mid+,r,dep+);
 }
 ll dis(point a,point b)
 {
     ll ans=;
     for(int i=;i<;++i) ans+=sqr(a.x[i]-b.x[i]);
     return ans;
 }
 ll ff(point x,int l,int r,int dep)
 {
     int cur=dep%;
     if(l>=r) {
         if(l==r&&x.c>=p[l].c) {
             ll cnt=dis(x,p[l]);
             if(p[l].c<=x.c&&(cnt<ans||(cnt==ans&&p[l].id<ansid)))
             {
                 ansid=p[l].id;
                 ans=cnt;
                 idd=l;
             }
         }
         return INF;
     }
     int mid=(l+r)>>;
     ll res=dis(x,p[mid]);
     if(p[mid].c<=x.c&&(res<ans||(res==ans&&p[mid].id<ansid)))
     {
         ans=res;ansid=p[mid].id;
         idd=mid;
     }
     ll tmp;
     if(x.x[cur]<p[mid].x[cur])
     {
         tmp=ff(x,l,mid-,dep+);
         if(ans>sqr(x.x[cur]-p[mid].x[cur])) ff(x,mid+,r,dep+);
     }
     else
     {
         tmp=ff(x,mid+,r,dep+);
         if(ans>sqr(x.x[cur]-p[mid].x[cur])) ff(x,l,mid-,dep+);
     }
 }
 int main()
 {
     scanf("%d",&T);
     while(T--)
     {
         scanf("%d%d",&n,&m);
         for(int i=;i<n;++i)
         {
             scanf("%d%d%d",&p[i].x[],&p[i].x[],&p[i].c);
             p[i].id=i;
         }
         build(,n-,);
         while(m--)
         {
             scanf("%d%d%d",&d.x[],&d.x[],&d.c);
             ans=INF;
             ff(d,,n-,);
             printf("%lld %lld %lld\n",p[idd].x[],p[idd].x[],p[idd].c);
         }
     }
 }