洛谷 P4357 [CQOI2016]K远点对【KD树】

题目描述：

$n$个点，求欧氏距离下的第$k$远点对的距离的平方。
$n<=100000,k<=\min (100,n(n-1)/2)$

题目分析：

一个堆维护前2k大（每个点对算了两次）
每个点在KD树中查询，取全局第2k大。
跟 求k远点 差不多。

Code：

#include<bits/stdc++.h>
#define maxn 100005
#define LL long long
using namespace std;
char cb[1<<18],*cs,*ct;
#define getc() (cs==ct&&(ct=(cs=cb)+fread(cb,1,1<<18,stdin),cs==ct)?0:*cs++)
inline void read(int &a){
	char c;bool f=0;while(!isdigit(c=getc())) c=='-'&&(f=1);
	for(a=c-'0';isdigit(c=getc());a=a*10+c-'0'); f&&(a=-a);	
}
const LL inf = 9223372036854775806ll;
int n,k,id[maxn],rt,X,Y;
struct node{
	int d[2],mn[2],mx[2],ch[2];
	void upd(node *t){
		mn[0]=mx[0]=d[0],mn[1]=mx[1]=d[1];
		for(int i=0;i<2;i++) if(ch[i]) for(int j=0;j<2;j++) mn[j]=min(mn[j],t[ch[i]].mn[j]),mx[j]=max(mx[j],t[ch[i]].mx[j]);	
	}
}t[maxn];
priority_queue<LL,vector<LL>,greater<LL> >S;
template<int D>bool cmp(int i,int j){return t[i].d[D]<t[j].d[D];}
bool (*fun[2])(int i,int j)={cmp<0>,cmp<1>};
void build(int &i,int l,int r,int D){
	if(l>r) {i=0;return;}
	int o=(l+r)>>1;nth_element(id+l,id+o,id+r+1,fun[D]),i=id[o];
	build(t[i].ch[0],l,o-1,D^1),build(t[i].ch[1],o+1,r,D^1);
	t[i].upd(t);
}
#define sqr(x) (1ll*(x)*(x))
inline LL G(int i){
	if(!i) return inf;
	return max(sqr(X-t[i].mn[0]),sqr(X-t[i].mx[0]))+max(sqr(Y-t[i].mn[1]),sqr(Y-t[i].mx[1]));
}
inline void ins(int i){
	LL now=sqr(X-t[i].d[0])+sqr(Y-t[i].d[1]);
	if(S.size()<k) S.push(now);
	else if(now>S.top()) S.pop(),S.push(now);
}
void query(int i){
	ins(i);
	LL tmp[2]={G(t[i].ch[0]),G(t[i].ch[1])};int p=tmp[1]>tmp[0];
	if(t[i].ch[p]&&(S.size()<k||tmp[p]>S.top())) query(t[i].ch[p]);
	p^=1;
	if(t[i].ch[p]&&(S.size()<k||tmp[p]>S.top())) query(t[i].ch[p]);
}
int main()
{
	read(n),read(k),k*=2;
	for(int i=1;i<=n;i++) id[i]=i,read(t[i].d[0]),read(t[i].d[1]);
	build(rt,1,n,0);
	for(int i=1;i<=n;i++) X=t[i].d[0],Y=t[i].d[1],query(rt);
	printf("%lld\n",S.top());
}