ble getc getch printf friend discuss ++ i++ esc

題解: 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

[BZOJ]2626: JZPFAR