UVA10245 The Closest Pair Problem(平面最近点对 分治裸题)

题目

给N(0<N<=1e4)个点，坐标范围[0,4e4]，

求N个点之间的最近点对的距离，

如果该距离大于1e4，输出INFINITY，否则输出距离

思路来源

《挑战程序设计竞赛》第二版

题解

先按x轴排增序，分治，考虑跨轴的贡献，

设当前左右两边已经递归出最小距离为d，

枚举中点的x值为当前中轴时，只考虑距中轴距离d以内的点，视为合法点

对于[l,r)区间，归并按y增序排序，

枚举每个合法点，在vector里向回检查距离当前合法点y轴距离在d以内的点，

并不断将合法点加入vector，能更新d就更新d

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e4+10;
const double INF=2e4,mx=1e4,eps=1e-6;
int n;
struct node
{
	double x,y;
}e[N],f[N];
bool cmp1(node a,node b)
{
	return a.x<b.x;
}
bool operator<(node a,node b)
{
	return a.y<b.y;
}
double cal(node a,node b)
{
	return sqrt((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y));
}
void merge(node *e,int l,int r)//归并 
{
	if(l+1>=r)return;
	int mid=(l+r+1)>>1,a=l,b=mid,c=0;//[l,mid)[mid,r)
	for(;a<mid||b<r;)
	{
		if(a<mid&&(b==r||e[a]<e[b]))f[c++]=e[a++];
		else f[c++]=e[b++];
	}
	for(int i=0;i<c;++i)
	e[l+i]=f[i];
}
double cdq(int l,int r)
{
	if(l+1>=r)return INF;
	int mid=(l+r+1)>>1;
	double x=e[mid].x;
	double d=min(cdq(l,mid),cdq(mid,r));
	merge(e,l,r);//在本区间内按y增序归并 
	vector<node>b;
	for(int i=l;i<r;++i)
	{
		if(fabs(e[i].x-x)>=d)continue;//距轴超过d 
		for(int j=0;j<b.size();++j)//按y增序排 先往回找近的 
		{
			double dx=e[i].x-b[b.size()-1-j].x;
			double dy=e[i].y-b[b.size()-1-j].y;
			if(dy>=d)break;//向回找 y距离超过d即break 
			d=min(d,sqrt(dx*dx+dy*dy)); 
		}
		b.push_back(e[i]);
	}
	return d;
}
int main()
{
	while(~scanf("%d",&n)&&n)
	{
		for(int i=0;i<n;++i)
		scanf("%lf%lf",&e[i].x,&e[i].y);
		sort(e,e+n,cmp1);
		double ans=cdq(0,n);
		if(ans>mx+eps)puts("INFINITY");
		else printf("%.4lf\n",ans);
	}
	return 0;
}