[BZOJ]2626: JZPFAR

题解:   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 MB
Submit: 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，表示一个询问。

Output

　　m行，每行一个整数，表示相应的询问的答案。

Sample Input

3
0 0
0 1
0 2
3
1 1 2
0 0 3
0 1 1

Sample Output

3
1
1