KDT玄学复杂度能过系列……
维护一个堆存全局前k远,枚举点对里的一个点在kdt上爆搜
加两个剪,一个是如果到一个矩形里任意一个点都不能更新答案就不搜,还有就是两个子树里先搜到这个点最远距离远的
由于每个点对会被算两次,k要乘2
#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<bitset>
using namespace std;
#define MAXN 100010
#define MAXM 1010
#define ll long long
#define INF 100000000000000000
#define MOD 1000000007
#define eps 1e-8
struct pt{
ll x;
ll y;
pt(){
}
pt(ll _x,ll _y){
x=_x;
y=_y;
}
int num;
};
pt v[MAXN],p[MAXN],L[MAXN],R[MAXN];
int son[MAXN][2];
int tot;
int n,k;
int now;
int rt;
priority_queue<ll>q;
ll dis(pt x,pt y){
return ((x.x-y.x)*(x.x-y.x)+(x.y-y.y)*(x.y-y.y));
}
bool cmpx(pt x,pt y){
return x.x<y.x;
}
bool cmpy(pt x,pt y){
return x.y<y.y;
}
bool (*cmp[])(pt x,pt y)={cmpx,cmpy};
inline void ud(int x){
L[x]=R[x]=v[x];
if(son[x][0]){
L[x].x=min(L[x].x,L[son[x][0]].x);
L[x].y=min(L[x].y,L[son[x][0]].y);
R[x].x=max(R[x].x,R[son[x][0]].x);
R[x].y=max(R[x].y,R[son[x][0]].y);
}
if(son[x][1]){
L[x].x=min(L[x].x,L[son[x][1]].x);
L[x].y=min(L[x].y,L[son[x][1]].y);
R[x].x=max(R[x].x,R[son[x][1]].x);
R[x].y=max(R[x].y,R[son[x][1]].y);
}
}
void build(int &x,int y,int z,bool f){
if(y>z){
return ;
}
x=++tot;
int mid=y+z>>1;
nth_element(p+y,p+mid,p+z+1,cmp[f]);
v[x]=p[mid];
build(son[x][0],y,mid-1,f^1);
build(son[x][1],mid+1,z,f^1);
ud(x);
}
void dfs(int x){
ll d=dis(v[x],p[now]);
if(q.size()<k){
q.push(-d);
}else if(d>-q.top()){
q.pop();
q.push(-d);
}
ll d1,d2;
if(son[x][0]){
d1=dis(p[now],pt(L[son[x][0]].x,L[son[x][0]].y));
d1=max(d1,dis(p[now],pt(L[son[x][0]].x,R[son[x][0]].y)));
d1=max(d1,dis(p[now],pt(R[son[x][0]].x,L[son[x][0]].y)));
d1=max(d1,dis(p[now],pt(R[son[x][0]].x,R[son[x][0]].y)));
}
if(son[x][1]){
d2=dis(p[now],pt(L[son[x][1]].x,L[son[x][1]].y));
d2=max(d2,dis(p[now],pt(L[son[x][1]].x,R[son[x][1]].y)));
d2=max(d2,dis(p[now],pt(R[son[x][1]].x,L[son[x][1]].y)));
d2=max(d2,dis(p[now],pt(R[son[x][1]].x,R[son[x][1]].y)));
}
if(son[x][0]&&son[x][1]){
if(d1>d2){
if(q.size()<k||(d1>-q.top())){
dfs(son[x][0]);
}
if(q.size()<k||(d2>-q.top())){
dfs(son[x][1]);
}
}else{
if(q.size()<k||(d2>-q.top())){
dfs(son[x][1]);
}
if(q.size()<k||(d1>-q.top())){
dfs(son[x][0]);
}
}
}else if(son[x][0]){
if(q.size()<k||(d1>-q.top())){
dfs(son[x][0]);
}
}else if(son[x][1]){
if(q.size()<k||(d2>-q.top())){
dfs(son[x][1]);
}
}
}
int main(){
int i;
scanf("%d%d",&n,&k);
for(i=1;i<=n;i++){
scanf("%lld%lld",&p[i].x,&p[i].y);
}
build(rt,1,n,0);
k=k*2-1;
for(i=1;i<=n;i++){
now=i;
dfs(rt);
}
printf("%lld\n",-q.top());
return 0;
}
/*
*/