bzoj4520 [Cqoi2016]K远点对（K-D tree）

类似bzoj3053，求第K远点对，我们暴力的循环n个点，每次拿着这个点去找能更新答案队列的点。因为每个点对我们都计算了两次，所以要开两倍k的队列，最后top就是答案。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x7fffffff
#define N 100010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,k,rt,D;
inline ll sqr(ll x){return x*x;}
struct point{
    int d[2];
    int& operator[](int x){return d[x];}
    friend bool operator<(point a,point b){return a[D]<b[D];}
    friend ll dis(point a,point b){return sqr(a[0]-b[0])+sqr(a[1]-b[1]);}
}P[N],T;
struct node{
    point x;int lc,rc,mx[2],mn[2];
}tr[N];priority_queue<ll,vector<ll>,greater<ll> >q;
inline void update(int p){
    int l=tr[p].lc,r=tr[p].rc;
    for(int i=0;i<2;++i){
        tr[p].mn[i]=min(tr[p].mn[i],min(tr[l].mn[i],tr[r].mn[i]));
        tr[p].mx[i]=max(tr[p].mx[i],max(tr[l].mx[i],tr[r].mx[i]));
    }
}
inline void build(int &p,int l,int r,int op){
    int mid=l+r>>1;p=mid;D=op;
    nth_element(P+l,P+mid,P+r+1);tr[p].x=P[mid];
    for(int i=0;i<2;++i) tr[p].mn[i]=tr[p].mx[i]=tr[p].x[i];
    if(l<mid) build(tr[p].lc,l,mid-1,op^1);
    if(r>mid) build(tr[p].rc,mid+1,r,op^1);update(p);
}
inline ll calc(int p){
    if(!p) return 0;
    ll res=0;
    for(int i=0;i<2;++i) res+=max(sqr(tr[p].mn[i]-T[i]),sqr(tr[p].mx[i]-T[i]));
    return res;
}
inline void ask(int p){
    ll tmp=dis(tr[p].x,T);
    if(tmp>q.top()) q.pop(),q.push(tmp);
    ll dl=calc(tr[p].lc),dr=calc(tr[p].rc);
    if(dl>dr){
        if(dl>q.top()) ask(tr[p].lc);
        if(dr>q.top()) ask(tr[p].rc);
    }else{
        if(dr>q.top()) ask(tr[p].rc);
        if(dl>q.top()) ask(tr[p].lc);
    }
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();k=read()<<1;
    for(int i=0;i<2;++i) tr[0].mn[i]=inf,tr[0].mx[i]=-inf;
    for(int i=1;i<=n;++i) P[i][0]=read(),P[i][1]=read();
    build(rt,1,n,0);
    for(int i=1;i<=k;++i) q.push(0);
    for(int i=1;i<=n;++i){T=P[i];ask(rt);}
    printf("%lld\n",q.top());
    return 0;
}