bzoj4520 [Cqoi2016]K远点对

http://www.elijahqi.win/2018/01/12/bzoj4520-cqoi2016k%e8%bf%9c%e7%82%b9%e5%af%b9/Description

已知平面内 N 个点的坐标，求欧氏距离下的第 K 远点对。

Input
输入文件第一行为用空格隔开的两个整数 N， K。接下来 N 行，每行两个整数 X,Y，表示一个点
的坐标。1 < = N < = 100000， 1 < = K < = 100， K < = N*(N−1)/2 ， 0 < = X, Y < 2^31。

Output

输出文件第一行为一个整数，表示第 K 远点对的距离的平方（一定是个整数）。

Sample Input
10 5
0 0
0 1
1 0
1 1
2 0
2 1
1 2
0 2
3 0
3 1
Sample Output
9
这题和上午写的bzoj3053很像 这题求的是平面内第k远的点 上一题怎么搞 开个优先队列搞啊‘这题也是一样我最暴力的做法就是o(n^2)枚举 然后验证 那么不妨我也这么做 开一个小根堆 每次都在我建出的kd-tree上去跑一下 求出我这个从前面1~i的最远的k个点 在一个节点上时如果这个点大于我的最小值 那么我就去做 剩下的同之前一样 可以看代码 注意 queue中需要存2*k个元素 why 因为 我是n^2个点对 所以每个都*2了 最后直接输出堆首即可

#include<cmath>
#include<queue>
#include<cstdio>
#include<algorithm>
#define N 110000
#define inf 0x3f3f3f3f
#define ll long long
using namespace std;
inline char gc(){
    static char now[1<<16],*S,*T;
    if (T==S){T=(S=now)+fread(now,1,1<<16,stdin);if (T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0;char ch=gc();
    while(ch<'0'||ch>'9') ch=gc();
    while(ch<='9'&&ch>='0')x=x*10+ch-'0',ch=gc();
    return x;
}
int n,k,D,root; 
struct node1{
    int d[2];
    int& operator[](int x){return d[x];}
    friend bool operator<(node1 a,node1 b){return a[D]<b[D];} 
}point[N],P;
struct node{
    node1 x;int left,right,min[2],max[2];
}tree[N];
priority_queue<ll,vector<ll>,greater<ll> >q;
inline void update(int x){
    int l=tree[x].left,r=tree[x].right;
    for (int i=0;i<2;++i) tree[x].min[i]=min(tree[x].min[i],min(tree[l].min[i],tree[r].min[i]));
    for (int i=0;i<2;++i) tree[x].max[i]=max(tree[x].max[i],max(tree[l].max[i],tree[r].max[i]));
}
inline void build(int &x,int l,int r,int dim){
    int mid=l+r>>1;x=mid;nth_element(point+l,point+mid,point+r+1);
    tree[x].x=point[mid];for (int i=0;i<2;++i) tree[x].min[i]=tree[x].max[i]=point[mid][i];
    if (l<mid) build(tree[x].left,l,mid-1,dim^1);
    if (r>mid) build(tree[x].right,mid+1,r,dim^1);update(x);
}
inline ll calc(node1 a){
    ll tmp=0;for (int i=0;i<2;++i) tmp+=(ll) (a[i]-P[i])*(a[i]-P[i]);return tmp;
}
inline ll calc1(int x){
    ll tmp1,tmp2;
    tmp1=max(abs(P[0]-tree[x].min[0]),abs(tree[x].max[0]-P[0]));
    tmp2=max(abs(P[1]-tree[x].min[1]),abs(tree[x].max[1]-P[1]));return tmp1*tmp1+tmp2*tmp2;
}
inline void query(int x){
    ll tmp=calc(tree[x].x);if (q.top()<tmp) q.pop(),q.push(tmp);
    ll disl=0,disr=0;if (tree[x].left) disl=calc1(tree[x].left);if (tree[x].right) disr=calc1(tree[x].right);
    if(disl>disr){
        if (disl>q.top()) query(tree[x].left);
        if (disr>q.top()) query(tree[x].right);
    }else{
        if (disr>q.top()) query(tree[x].right);
        if (disl>q.top()) query(tree[x].left);
    }
}
int main(){
    freopen("bzoj4520.in","r",stdin);
    n=read();k=read();k<<=1;for (int i=0;i<2;++i) tree[0].min[i]=tree[0].max[i]=inf;
    for (int i=1;i<=n;++i) for (int j=0;j<2;++j) point[i][j]=read();
    build(root,1,n,0);for (int i=1;i<=k;++i) q.push(0);
    for (int i=1;i<=n;++i){
        P=point[i];query(root);
    }printf("%lld\n",q.top());
    return 0;
}