【bzoj4520】 Cqoi2016—K远点对

http://www.lydsy.com/JudgeOnline/problem.php?id=4520 (题目链接)

题意

　　求平面内第K远点对的距离。

Solution

　　左转题解：jump

细节

　　刚开始我还开了两个堆，想想其实是没必要的→_→

　　距离什么的开LL

代码

// bzoj4520

#include<algorithm>

#include<iostream>

#include<cstdlib>

#include<cstring>

#include<cstdio>

#include<vector>

#include<cmath>

#include<queue>

#define LL long long

#define inf 1<<30

#define Pi acos(-1.0)

#define free(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);

using namespace std;

const int maxn=100010;

int n,m,D,K,rt;

priority_queue<LL,vector<LL>,greater<LL> > q;

struct KDtree {

	int l,r,v[2],mn[2],mx[2];

	friend bool operator < (const KDtree a,const KDtree b) {

		return a.v[D]<b.v[D];

	}

}tr[maxn],S;

void update(int k) {

	for (int i=0;i<=1;i++) {

		if (tr[k].l) {

			tr[k].mn[i]=min(tr[k].mn[i],tr[tr[k].l].mn[i]);

			tr[k].mx[i]=max(tr[k].mx[i],tr[tr[k].l].mx[i]);

		}

		if (tr[k].r) {

			tr[k].mn[i]=min(tr[k].mn[i],tr[tr[k].r].mn[i]);

			tr[k].mx[i]=max(tr[k].mx[i],tr[tr[k].r].mx[i]);

		}

	}

}

int build(int l,int r,int p) {

	D=p;

	int mid=(l+r)>>1;

	nth_element(tr+l,tr+mid,tr+r+1);

	if (l<mid) tr[mid].l=build(l,mid-1,p^1);

	if (r>mid) tr[mid].r=build(mid+1,r,p^1);

	update(mid);

	return mid;

}

LL dis(LL x1,LL y1,LL x2,LL y2) {

	return (x1-x2)*(x1-x2)+(y1-y2)*(y1-y2);

}

LL eva(int k) {

    LL lu=dis(S.v[0],S.v[1],tr[k].mn[0],tr[k].mn[1]);

    LL ru=dis(S.v[0],S.v[1],tr[k].mx[0],tr[k].mn[1]);

    LL ld=dis(S.v[0],S.v[1],tr[k].mn[0],tr[k].mx[1]);

    LL rd=dis(S.v[0],S.v[1],tr[k].mx[0],tr[k].mx[1]);

    return max(max(lu,ru),max(ld,rd));

}

void query(int k) {

	if (!k) return;

	LL d=dis(S.v[0],S.v[1],tr[k].v[0],tr[k].v[1]);

	if (q.top()<d) q.pop(),q.push(d);

	LL dl=eva(tr[k].l),dr=eva(tr[k].r);

	if (dl>dr) {

		if (q.top()<dl) query(tr[k].l);

		if (q.top()<dr) query(tr[k].r);

	}

	else {

		if (q.top()<dr) query(tr[k].r);

		if (q.top()<dl) query(tr[k].l);

	}

}

int main() {

	scanf("%d%d",&n,&K);

	for (int i=1;i<=n;i++) {

		scanf("%d%d",&tr[i].v[0],&tr[i].v[1]);

		tr[i].mn[0]=tr[i].mx[0]=tr[i].v[0];

		tr[i].mn[1]=tr[i].mx[1]=tr[i].v[1];

	}

	rt=build(1,n,0);

	for (int i=1;i<=2*K;i++) q.push(0);

	for (int i=1;i<=n;i++) S=tr[i],query(rt);

	printf("%lld",q.top());

	return 0;

}