BZOJ4520 [Cqoi2016]K远点对

KDT玄学复杂度能过系列……

维护一个堆存全局前k远,枚举点对里的一个点在kdt上爆搜

加两个剪,一个是如果到一个矩形里任意一个点都不能更新答案就不搜,还有就是两个子树里先搜到这个点最远距离远的

由于每个点对会被算两次,k要乘2

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
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_queueq;
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.xz){
		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()-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()-q.top())){
				dfs(son[x][0]);
			}
			if(q.size()-q.top())){
				dfs(son[x][1]);
			}
		}else{
			if(q.size()-q.top())){
				dfs(son[x][1]);
			}
			if(q.size()-q.top())){
				dfs(son[x][0]);
			}
		}
	}else if(son[x][0]){
		if(q.size()-q.top())){
			dfs(son[x][0]);
		}
	}else if(son[x][1]){
		if(q.size()-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;
}

/*

*/


你可能感兴趣的:(BZOJ,KDT)