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

 1 #include<bits/stdc++.h>
 2 typedef long long ll;
 3 using namespace std;
 4 const int N=2e5+5;
 5 const ll INF=1e18;
 6 int T,n,m;
 7 struct point 
 8 {
 9     ll x[2];
10     int c,id;
11 }p[N],d;
12 ll idd,ansid,ans;
13 ll sqr(ll x) {return x*x;}
14 int cur;
15 bool cmp(point a,point b) 
16 {
17     return a.x[cur]<b.x[cur];
18 }
19 void build(int l,int r,int dep)
20 {
21     if(l>=r) return;
22     int mid=(l+r)>>1;
23     cur=dep%2;
24     nth_element(p+l,p+mid,p+r+1,cmp);
25     build(l,mid-1,dep+1);
26     build(mid+1,r,dep+1);
27 }
28 ll dis(point a,point b) 
29 {
30     ll ans=0;
31     for(int i=0;i<2;++i) ans+=sqr(a.x[i]-b.x[i]);
32     return ans;
33 }
34 ll ff(point x,int l,int r,int dep) 
35 {
36     int cur=dep%2;
37     if(l>=r) {
38         if(l==r&&x.c>=p[l].c) {
39             ll cnt=dis(x,p[l]);
40             if(p[l].c<=x.c&&(cnt<ans||(cnt==ans&&p[l].id<ansid))) 
41             {
42                 ansid=p[l].id;
43                 ans=cnt;
44                 idd=l;
45             }
46         }
47         return INF;
48     }
49     int mid=(l+r)>>1;
50     ll res=dis(x,p[mid]);
51     if(p[mid].c<=x.c&&(res<ans||(res==ans&&p[mid].id<ansid))) 
52     {
53         ans=res;ansid=p[mid].id;
54         idd=mid;
55     }
56     ll tmp;
57     if(x.x[cur]<p[mid].x[cur]) 
58     {
59         tmp=ff(x,l,mid-1,dep+1);
60         if(ans>sqr(x.x[cur]-p[mid].x[cur])) ff(x,mid+1,r,dep+1);
61     }
62     else 
63     {
64         tmp=ff(x,mid+1,r,dep+1);
65         if(ans>sqr(x.x[cur]-p[mid].x[cur])) ff(x,l,mid-1,dep+1);
66     }
67 }
68 int main() 
69 {
70     scanf("%d",&T);
71     while(T--) 
72     {
73         scanf("%d%d",&n,&m);
74         for(int i=0;i<n;++i) 
75         {
76             scanf("%d%d%d",&p[i].x[0],&p[i].x[1],&p[i].c);
77             p[i].id=i;
78         }
79         build(0,n-1,0);
80         while(m--) 
81         {
82             scanf("%d%d%d",&d.x[0],&d.x[1],&d.c);
83             ans=INF;
84             ff(d,0,n-1,0);
85             printf("%lld %lld %lld\n",p[idd].x[0],p[idd].x[1],p[idd].c);
86         }
87     }
88 }