题解: KDtree裸题 用堆维护离得最近的K个点 就行了
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <stack>
#include <queue>
#include <cmath>
#include <set>
#include <map>
#define mp make_pair
#define pb push_back
#define pii pair<int,int>
#define link(x) for(edge *j=h[x];j;j=j->next)
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,r,l) for(int i=r;i>=l;i--)
const int MAXN=1e5+10;
const double eps=1e-8;
#define ll long long
using namespace std;
struct edge{int t,v;edge*next;}e[MAXN<<1],*h[MAXN],*o=e;
void add(int x,int y,int vul){o->t=y;o->v=vul;o->next=h[x];h[x]=o++;}
ll read(){
ll x=0,f=1;char ch=getchar();
while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
return x*f;
}
int n,m,k,d,rt;
typedef struct node{
int p[2],minn[2],maxx[2],ch[2],id;
friend bool operator<(node aa,node bb){
if(aa.p[d]!=bb.p[d])return aa.p[d]<bb.p[d];
return aa.p[d^1]<bb.p[d^1];
}
}node;
node a[MAXN];
void up(int x,int y){
inc(i,0,1)a[x].minn[i]=min(a[x].minn[i],a[y].minn[i]),a[x].maxx[i]=max(a[x].maxx[i],a[y].maxx[i]);
}
int built(int l,int r,int now){
int mid=(l+r)>>1;
d=now;nth_element(a+l,a+mid,a+r+1);
inc(i,0,1)a[mid].minn[i]=a[mid].maxx[i]=a[mid].p[i],a[mid].ch[i]=0;
if(l<mid)a[mid].ch[0]=built(l,mid-1,now^1),up(mid,a[mid].ch[0]);
if(r>mid)a[mid].ch[1]=built(mid+1,r,now^1),up(mid,a[mid].ch[1]);
return mid;
}
typedef struct Tmp{
ll dis;int id;
friend bool operator<(Tmp aa,Tmp bb){
if(aa.dis==bb.dis)return aa.id<bb.id;
return aa.dis>bb.dis;
}
}Tmp;
priority_queue<Tmp>que;
node D;
ll get_ans(int x){
ll ans=0;
inc(i,0,1)ans+=max(1ll*(D.p[i]-a[x].maxx[i])*(D.p[i]-a[x].maxx[i]),1ll*(D.p[i]-a[x].minn[i])*(D.p[i]-a[x].minn[i]));
return ans;
}
ll dist(node x,node y){
return 1ll*(x.p[0]-y.p[0])*(x.p[0]-y.p[0])+1ll*(x.p[1]-y.p[1])*(x.p[1]-y.p[1]);
}
void query(int x){
if(!x)return ;
ll t=dist(D,a[x]);
if(t>=(que.top()).dis){
if(t==(que.top()).dis&&a[x].id<(que.top()).id)que.pop(),que.push((Tmp){t,a[x].id});
else{
if(t>(que.top()).dis)que.pop(),que.push((Tmp){t,a[x].id});
}
}
ll t1=(a[x].ch[0]?get_ans(a[x].ch[0]):-1);
ll t2=(a[x].ch[1]?get_ans(a[x].ch[1]):-1);
if(t1>t2){
if(t1>=(que.top()).dis)query(a[x].ch[0]);
if(t2>=(que.top()).dis)query(a[x].ch[1]);
}
else{
if(t2>=(que.top()).dis)query(a[x].ch[1]);
if(t1>=(que.top()).dis)query(a[x].ch[0]);
}
}
int main(){
n=read();
inc(i,1,n){
a[i].p[0]=read();a[i].p[1]=read();a[i].id=i;
}
rt=built(1,n,0);
m=read();
while(m--){
D.p[0]=read();D.p[1]=read();k=read();
inc(i,1,k)que.push((Tmp){-1,0});
query(rt);
printf("%d\n",(que.top()).id);
inc(i,1,k)que.pop();
}
return 0;
}
2626: JZPFAR
Time Limit: 50 Sec Memory Limit: 128 MBSubmit: 1291 Solved: 505
[Submit][Status][Discuss]
Description
平面上有n个点。现在有m次询问,每次给定一个点(px, py)和一个整数k,输出n个点中离(px, py)的距离第k大的点的标号。如果有两个(或多个)点距离(px, py)相同,那么认为标号较小的点距离较大。
Input
第一行,一个整数n,表示点的个数。
下面n行,每行两个整数x_i, y_i,表示n个点的坐标。点的标号按照输入顺序,分别为1..n。
下面一行,一个整数m,表示询问个数。
下面m行,每行三个整数px_i, py_i, k_i,表示一个询问。
下面n行,每行两个整数x_i, y_i,表示n个点的坐标。点的标号按照输入顺序,分别为1..n。
下面一行,一个整数m,表示询问个数。
下面m行,每行三个整数px_i, py_i, k_i,表示一个询问。
Output
m行,每行一个整数,表示相应的询问的答案。
Sample Input
3
0 0
0 1
0 2
3
1 1 2
0 0 3
0 1 1
0 0
0 1
0 2
3
1 1 2
0 0 3
0 1 1
Sample Output
3
1
1
1
1