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;

}

 

你可能感兴趣的:(poj)