【BZOJ】【2626】JZPFAR

KD-Tree

---

　　0.0找第k大……

　　裸KD-Tree……跟之前那道找最近的k个点大同小异

　　一开始理解错：第K大是第K远……不是第K近……（Tunix你个sb

　　感觉容易出错的是0号点= =边界情况需要仔细处理……根据题意而定的，比如这题就必须将0号点的距离设置成最近……比如-2……（因为我一开始向堆里加的占位点的距离是-1

/**************************************************************
    Problem: 2626
    User: Tunix
    Language: C++
    Result: Accepted
    Time:12956 ms
    Memory:4796 kb
****************************************************************/
 
//BZOJ 2626
#include<queue>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define rep(i,n) for(int i=0;i<n;++i)
#define F(i,j,n) for(int i=j;i<=n;++i)
#define D(i,j,n) for(int i=j;i>=n;--i)
#define pb push_back
#define sqr(x) (x)*(x)
using namespace std;
typedef long long LL;
inline int getint(){
    int r=1,v=0; char ch=getchar();
    for(;!isdigit(ch);ch=getchar()) if (ch=='-') r=-1;
    for(; isdigit(ch);ch=getchar()) v=v*10-'0'+ch;
    return r*v;
}
const int N=100010,INF=1e9;
/*******************template********************/
int n,m,root,D;
struct Poi{
    int d[2],mn[2],mx[2],l,r,num;
    int& operator [] (int x){return d[x];}
    void read(){d[0]=getint();d[1]=getint();}
}t[N],now;
 
bool operator < (Poi a,Poi b) {return a[D]<b[D];}
struct node{
    LL dis;int num;
    node(LL d=0,int n=0):dis(d),num(n){}
};
bool operator < (const node &a,const node &b){
    return a.dis>b.dis || (a.dis==b.dis && a.num<b.num);
}
#define L t[o].l
#define R t[o].r
#define mid (l+r>>1)
void Push_up(int o){
    F(i,0,1){
        t[o].mn[i]=min(t[o].mn[i],min(t[L].mn[i],t[R].mn[i]));
        t[o].mx[i]=max(t[o].mx[i],max(t[L].mx[i],t[R].mx[i]));
    }
}
int build(int l,int r,int dir){
    D=dir;
    nth_element(t+l,t+mid,t+r+1);
    int o=mid;
    F(i,0,1) t[o].mn[i]=t[o].mx[i]=t[o][i];
    if (l<mid) L=build(l,mid-1,dir^1);
    if (r>mid) R=build(mid+1,r,dir^1);
    Push_up(o);
    return o;
}
 
inline LL dis(Poi a,Poi b){return (LL)sqr(a[0]-b[0])+(LL)sqr(a[1]-b[1]);}
inline LL getdis(int o){
    if (!o) return -2;
    LL ans=0;
    F(i,0,1)
        ans+=max((LL)sqr(t[o].mx[i]-now[i]),(LL)sqr(t[o].mn[i]-now[i]));
    return ans;
}
priority_queue<node>Q;
 
void query(int o){
    if (!o) return;
    LL dl=getdis(L),dr=getdis(R),d0=dis(t[o],now);
    if (d0>Q.top().dis || (d0==Q.top().dis && t[o].num<Q.top().num)){
        Q.pop(); Q.push(node(d0,t[o].num));
    }
    if (dl>dr){
        if (dl>=Q.top().dis) query(L);
        if (dr>=Q.top().dis) query(R);
    }else{
        if (dr>=Q.top().dis) query(R);
        if (dl>=Q.top().dis) query(L);
    }
}
 
int main(){
#ifndef ONLINE_JUDGE
    freopen("2626.in","r",stdin);
    freopen("2626.out","w",stdout);
#endif
    F(i,0,1) t[0].mn[i]=INF,t[0].mx[i]=-INF;
    n=getint();
    F(i,1,n) t[i].read(),t[i].num=i;
    root=build(1,n,1);
    int T=getint();
    while(T--){
        now.read(); int k=getint();
//      printf("%d %d %d
",now[0],now[1],k);
        while(!Q.empty()) Q.pop();
        F(i,1,k) Q.push(node(-1,0));
        query(root);
        printf("%d
",Q.top().num);
//      printf("Q.size()=%d
",Q.size());
//      F(i,1,k)
//          printf("%d -- %lld %lld
",i,Q.top().dis,Q.top().num),Q.pop();
//      puts("");
    }
    return 0;
}

2626: JZPFAR

Time Limit: 50 Sec  Memory Limit: 128 MB
Submit: 665  Solved: 248
[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

数据规模和约定
　　50%的数据中，n个点的坐标在某范围内随机分布。
　　100%的数据中，n<=10^5, m<=10^4, 1<=k<=20，所有点(包括询问的点)的坐标满足绝对值<=10^9，n个点中任意两点坐标不同，m个询问的点的坐标在某范围内随机分布。

HINT

Source

[Submit][Status][Discuss]