【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=,v=; char ch=getchar();
     for(;!isdigit(ch);ch=getchar()) if (ch=='-') r=-;
     for(; isdigit(ch);ch=getchar()) v=v*-''+ch;
     return r*v;
 }
 const int N=,INF=1e9;
 /*******************template********************/
 int n,m,root,D;
 struct Poi{
     int d[],mn[],mx[],l,r,num;
     int& operator [] (int x){return d[x];}
     void read(){d[]=getint();d[]=getint();}
 }t[N],now;

 bool operator < (Poi a,Poi b) {return a[D]<b[D];}
 struct node{
     LL dis;int num;
     node(LL d=,int n=):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,,){
         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+);
     int o=mid;
     F(i,,) t[o].mn[i]=t[o].mx[i]=t[o][i];
     if (l<mid) L=build(l,mid-,dir^);
     if (r>mid) R=build(mid+,r,dir^);
     Push_up(o);
     return o;
 }

 inline LL dis(Poi a,Poi b){return (LL)sqr(a[]-b[])+(LL)sqr(a[]-b[]);}
 inline LL getdis(int o){
     if (!o) return -;
     LL ans=;
     F(i,,)
         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,,) t[].mn[i]=INF,t[].mx[i]=-INF;
     n=getint();
     F(i,,n) t[i].read(),t[i].num=i;
     root=build(,n,);
     int T=getint();
     while(T--){
         now.read(); int k=getint();
 //      printf("%d %d %d\n",now[0],now[1],k);
         while(!Q.empty()) Q.pop();
         F(i,,k) Q.push(node(-,));
         query(root);
         printf("%d\n",Q.top().num);
 //      printf("Q.size()=%d\n",Q.size());
 //      F(i,1,k)
 //          printf("%d -- %lld %lld\n",i,Q.top().dis,Q.top().num),Q.pop();
 //      puts("");
     }
     return ;
 }

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]