解题：洛谷2093 JZPFAR

题面

初见K-D Tree

其实这样的题(欧几里得距离第$x$近点对)不应该用K-D Tree做，因为会被构造数据卡成$O(n^2)$，随机的另说。

但是并没有找到合适的K-D Tree的题(区域统计)，于是就凑活着写了，代码极丑预警

 // luogu-judger-enable-o2
 #include<queue>
 #include<cstdio>
 #include<cctype>
 #include<cstring>
 #include<algorithm>
 using namespace std;
 const int N=;
 const long long inf=1e9;
 struct a
 {
     int id;
     long long ds;
 };
 bool operator < (a x,a y)
 {
     return x.ds==y.ds?x.id<y.id:x.ds>y.ds;
 }
 priority_queue<a> hp;
 struct b
 {
     int ps[],mn[],mx[],sn[],id,idn;
 }kdt[N],qry;
 int n,m,k,r,f,typ;
 inline void read(int &x)
 {
     x=,f=; char ch=getchar();
     while(!isdigit(ch))
         f|=ch=='-',ch=getchar();
     while(isdigit(ch))
         x=(x<<)+(x<<)+(ch^),ch=getchar();
     x=f?-x:x;
 }
 void write(int x)
 {
     if(x>) write(x/);
     putchar(x%^);
 }
 bool operator < (b x,b y)
 {
     return x.ps[typ]<y.ps[typ];
 }
 void Pushup(int nde)
 {
     kdt[nde].idn=kdt[nde].id;
     int lson=kdt[nde].sn[],rson=kdt[nde].sn[];
     b ls=kdt[lson],rs=kdt[rson];
     if(lson) kdt[nde].idn=min(kdt[nde].idn,ls.idn);
     if(rson) kdt[nde].idn=min(kdt[nde].idn,rs.idn);
     for(int i=;i<=;i++)
     {
         kdt[nde].mn[i]=kdt[nde].mx[i]=kdt[nde].ps[i];
         if(lson) kdt[nde].mn[i]=min(kdt[nde].mn[i],ls.mn[i]),kdt[nde].mx[i]=max(kdt[nde].mx[i],ls.mx[i]);
         if(rson) kdt[nde].mn[i]=min(kdt[nde].mn[i],rs.mn[i]),kdt[nde].mx[i]=max(kdt[nde].mx[i],rs.mx[i]);
     }
 }
 int Create(int l,int r,int k)
 {
     typ=k; int mid=(l+r)/;
     nth_element(kdt+l,kdt+mid,kdt+r+);
     if(l<mid) kdt[mid].sn[]=Create(l,mid-,k^);
     if(r>mid) kdt[mid].sn[]=Create(mid+,r,k^);
     Pushup(mid); return mid;
 }
 long long calc1(int nde)
 {
     long long mnx=kdt[nde].mn[]-qry.ps[],mxx=kdt[nde].mx[]-qry.ps[];
     long long mny=kdt[nde].mn[]-qry.ps[],mxy=kdt[nde].mx[]-qry.ps[];
     return max(mnx*mnx,mxx*mxx)+max(mny*mny,mxy*mxy);
 }
 long long calc2(int nde)
 {
     long long dsx=kdt[nde].ps[]-qry.ps[];
     long long dsy=kdt[nde].ps[]-qry.ps[];
     return dsx*dsx+dsy*dsy;
 }
 inline bool farther(long long x,long long y)
 {
     a t=hp.top();
     return x>t.ds||(x==t.ds&&y<t.id);
 }
 inline bool further(long long x,long long y)
 {
     a t=hp.top();
     return x>t.ds||(x==t.ds&&kdt[y].idn<t.id);
 }
 void Query(int nde)
 {
     b tmp=kdt[nde]; long long dis=calc2(nde);
     int idx=tmp.id,lson=tmp.sn[],rson=tmp.sn[];
     if(farther(dis,idx))
         hp.pop(),hp.push((a){idx,dis});
     long long d1=lson?calc1(lson):-inf,d2=rson?calc1(rson):-inf;
     if(d1>d2)
     {
         if(further(d1,lson)) Query(lson);
         if(further(d2,rson)) Query(rson);
     }
     else
     {
         if(further(d2,rson)) Query(rson);
         if(further(d1,lson)) Query(lson);
     }
 }
 int main()
 {
     register int i;
     read(n);
     for(i=;i<=n;i++)
         read(kdt[i].ps[]),read(kdt[i].ps[]),kdt[i].id=i;
     r=Create(,n,),scanf("%d",&m);
     while(m--)
     {
         while(!hp.empty()) hp.pop();
         read(qry.ps[]),read(qry.ps[]),read(k);
         for(i=;i<=k;i++) hp.push((a){,-inf});
         Query(r),write(hp.top().id),puts("");
     }
     return ;
 }