POJ--3714[Raid] 分治发求最小点对

题意：

给你两个集合A，B(每个集合N个点)，要求输出一个最小点对（两点分别在A，B两个集合内）。其实就是分治法求最小点对的模板题，只不过在取最短距离时加上不同集合的限制即可。

 

感想：

但是郁闷的是，居然不停地TLE。。。。囧。。。查了一遍又一遍，最终发现，我一开始在get_dis（）函数中返回的不是两点的距离，而是两点的距离^2。本以为每次开方耗时较大，所以每次先不开方，最后把得到的ans开根号即可.....可恰恰是这样，它TLE了。。。不过还是不太知道为什么，可能是（因为这题X，Y的范围比较大1000000000，每次对这么大的数进行乘法比较耗时），唉，不知道不知道！！！

 

 

源代码：

/*分治算法求最小点对*/
/*AC代码：2172ms*/
#include <iostream>
#include <cmath>
#include <algorithm>
#define MAXN 200005

#define INF 1e100
using namespace std;
struct Point
{
	double x,y; 
	bool v;//用来区分两个集合
};
struct Point px[MAXN],py[MAXN];
double get_dis(Point p1,Point p2)
{
	if(p1.v==p2.v)//如果在同一个集合，返回INF
		return INF;
	return sqrt((p1.x-p2.x)*(p1.x-p2.x)+(p1.y-p2.y)*(p1.y-p2.y));
}
int cmpx(const void *p1,const void *p2)//用qsort一定要x，y都判断，不然TLE
{
	if(((struct Point *)p1)->x!=((struct Point *)p2)->x)
		return ((struct Point *)p1)->x>((struct Point *)p2)->x?1:-1;
	return ((struct Point *)p1)->y>((struct Point *)p2)->y?1:-1;
}
int cmpy(const void *p1,const void *p2)
{
	if(((struct Point *)p1)->y!=((struct Point *)p2)->y)
		return ((struct Point *)p1)->y>((struct Point *)p2)->y?1:-1;
	return ((struct Point *)p1)->x>((struct Point *)p2)->x?1:-1;
}
double min(double a,double b){return a<b?a:b;}
//-------核心代码------------//
double closest(int s,int e)
{
	if(s+1==e)
		return get_dis(px[s],px[e]);
	if(s+2==e)
		return min(get_dis(px[s],px[s+1]),min(get_dis(px[s+1],px[e]),get_dis(px[s],px[e])));
	int mid=(s+e)>>1;
	double ans=min(closest(s,mid),closest(mid+1,e));//递归求解
	int i,j,cnt=0;
	for(i=s;i<=e;i++)//把x坐标在px[mid].x-ans~px[mid].x+ans范围内的点取出来
	{
		if(px[i].x>=px[mid].x-ans&&px[i].x<=px[mid].x+ans)
			py[cnt++]=px[i];
	}
	qsort(py,cnt,sizeof(py[0]),cmpy);//按y坐标排序
	for(i=0;i<cnt;i++)
	{
		for(j=i+1;j<cnt;j++)//py数组中的点是按照y坐标升序的
		{
			if(py[j].y-py[i].y>=ans)
				break;
			ans=min(ans,get_dis(py[i],py[j]));
		}
	}
	return ans;
}
int main()
{
	int i,n,T;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d",&n);
		for(i=0;i<n;i++)
		{
			scanf("%lf%lf",&px[i].x,&px[i].y);
			px[i].v=true;
		}
		for(i=n;i<2*n;i++)
		{
			scanf("%lf%lf",&px[i].x,&px[i].y);
			px[i].v=false;
		}
		n<<=1;
		qsort(px,n,sizeof(px[0]),cmpx);
		double distance=closest(0,n-1);
		printf("%.3lf\n",distance);
	}
	return 0;
}