poj3714 最近点对

最近点对,采用分治方法。过程:
1对原数组依据x左标从小到大排序。
2二分数组,左边求出最小值,右边求出最小值,我们求最小的。
3找出对于左右两边的可能小于当前最小值的最近点对,更新最小值。
这题目需要区分一下点,让我们求的是闪兵到任意一个核电站的最短距离,加一个标志就可以了。
代码:

#include <iostream>
#include <stdio.h>
#include <iomanip>
#include <algorithm>
#include <cmath>
using namespace std;
const int N=200000+1;
const double INF=1e100;
struct Point{
	double x,y;
	bool flag;
};
Point m[N+1];
int tmp[N+1];
double dis(Point p1,Point p2)
{
	return sqrt((p1.x-p2.x)*(p1.x-p2.x)+(p1.y-p2.y)*(p1.y-p2.y));
}
double minL(double l1,double l2)
{
	return l1<=l2?l1:l2;
}
bool cmpy(int a,int b)
{
	return m[a].y<m[b].y;
}
bool cmpx(Point a,Point b)
{
	return a.x<b.x;
}
double getMinLen(Point *s,int left,int right)
{
	double rs=INF;
	if(left == right)
		return rs;
	if(left+1 == right)
	{
		if(s[left].flag == s[right].flag)
			return rs;
		return dis(s[left],s[right]);
	}
	int mid=(left+right)>>1;
	rs=getMinLen(s,left,mid);
	rs=minL(rs,getMinLen(s,mid+1,right));//rs现在是两边中最点值了
	int i,j,num=0;
	for(i=left;i<=right;++i)
	{
		if(fabs(s[i].x-s[mid].x)<=rs)//过滤掉不可能是最近点对的点
			tmp[num++]=i;
	}
	sort(tmp,tmp+num,cmpy);//按y坐标排序,减少比较次数
	double d=INF;
	for(i=0;i<num;++i)
	{
		for(j=i+1;j<num;++j)
		{
			if(fabs(s[tmp[i]].y-s[tmp[j]].y)>=rs)//i,j两点的y坐标已经大于rs,不需要在遍历了
				break;
			if(s[tmp[i]].flag!=s[tmp[j]].flag && (d=dis(s[tmp[i]],s[tmp[j]]))<rs)
				rs=d;
		}
	}
//	cout<<left<<" "<<right<<" "<<rs<<endl;
	return rs;
}
int main()
{
	int t,n,i;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d",&n);
		for(i=0;i<n;++i)
		{
			scanf("%lf%lf",&m[i].x,&m[i].y);
			m[i].flag=0;
		}
		for(i=0;i<n;++i)
		{
			scanf("%lf%lf",&m[i+n].x,&m[i+n].y);
			m[i+n].flag=1;
		}
		n<<=1;
		sort(m,m+n,cmpx);
		double ans=getMinLen(m,0,n-1);
		cout<<setiosflags(ios::fixed)<<setprecision(3)<<ans<<endl;
	}
	return 0;
}
原文地址:https://www.cnblogs.com/buptLizer/p/2241363.html