[算法竞赛进阶指南] 袭击 (最近点对)

题目

在与联盟的战斗中屡战屡败后，帝国撤退到了最后一个据点。

依靠其强大的防御系统，帝国击退了联盟的六波猛烈进攻。

经过几天的苦思冥想，联盟将军亚瑟终于注意到帝国防御系统唯一的弱点就是能源供应。

该系统由N个核电站供应能源，其中任何一个被摧毁都会使防御系统失效。

将军派出了N个特工进入据点之中，打算对能源站展开一次突袭。

不幸的是，由于受到了帝国空军的袭击，他们未能降落在预期位置。

作为一名经验丰富的将军，亚瑟很快意识到他需要重新安排突袭计划。

他现在最想知道的事情就是哪个特工距离其中任意一个发电站的距离最短。

你能帮他算出来这最短的距离是多少吗？

输入格式

输入中包含多组测试用例。

第一行输入整数T，代表测试用例的数量。

对于每个测试用例，第一行输入整数N。

接下来N行，每行输入两个整数X和Y，代表每个核电站的位置的X，Y坐标。

在接下来N行，每行输入两个整数X和Y，代表每名特工的位置的X，Y坐标。

输出格式

每个测试用例，输出一个最短距离值，结果保留三位小数。

每个输出结果占一行。

数据范围

1≤N≤100000,
0≤X,Y≤1000000000

输入样例

2
4
0 0
0 1
1 0
1 1
2 2
2 3
3 2
3 3
4
0 0
0 0
0 0
0 0
0 0
0 0
0 0
0 0

输出样例

1.414
0.000

分析：

• 最近点对分治解法的模板题。
• 只需要输入的时候用结构体存储点，里面再储存一个标记，求距离的时候，如果标记相同，距离就是无限远即可。
• 具体注释和实现可看代码

最近点对分治算法 详细分析可以看一看：https://blog.csdn.net/qq_41934571/article/details/97569782

代码：

#include <iostream>
#include <algorithm>
#include <cmath>
using namespace std;
const double INF=1e10;
const int N=100005;
 
int n;
struct node{
	double x,y;
	int flag;
	bool operator < (const node &a) const{
		return x<a.x;
	} 
}opt[N<<1],temp[N<<1];

double dis(node a,node b){//返回两点之间的距离 
	if(a.flag==b.flag) return INF;
	return sqrt(pow(abs(a.x-b.x),2)+pow(abs(a.y-b.y),2));
}

double solve(int l,int r){// 返回排序后下标l到r这个区间中最小点距
	//如果下标区间只有这一个点 返回最大值  
	if(l>=r) return INF;
	
	//二分递归，求两个左右区间的最小点距，并且将小的赋给ans存储 
	int mid=(l+r)>>1;
	double mid_x=opt[mid].x;
	double ans=min(solve(l,mid),solve(mid+1,r));
	
	//二分排序的合并过程  
	int left=l,right=mid+1,t=l;
	while(left<=mid && right<=r){
		if(opt[left].y<=opt[right].y) temp[t++]=opt[left++];
		else temp[t++]=opt[right++];
	}
	while(left<=mid) temp[t++]=opt[left++];
	while(right<=r) temp[t++]=opt[right++];
	for(int i=l;i<=r;i++){
		opt[i]=temp[i];
	}
	
	//从左到右将opt中的在mid_x-ans至mid_x+ans的点存进temp
	//这样temp中的点既是按y排序的，又是在mid_x-ans至mid_x+ans区间里的 
	int k=0;
	for(int i=l;i<=r;i++){
		if(opt[i].x>=mid_x-ans&&opt[i].x<=mid_x+ans){
			temp[k++]=opt[i];
		}
	}
	
	//从有序的temp数组中 找到每个点y坐标上面不超过ans的点的点距离求出
	//并且更新答案  
	for(int i=0;i<k;i++){
		for(int j=i-1;j>=0&&temp[i].y-temp[j].y<=ans;j--){
			ans=min(ans,dis(temp[i],temp[j]));
		}
	}
	
	//返回这个l到r下标的最小点距 
	return ans;
}

int main()
{
	int t;
	cin>>t;
	while(t--){
		cin>>n;
		for(int i=0;i<n;i++) {
			cin>>opt[i].x>>opt[i].y;
			opt[i].flag=0;
		}
		for(int i=n;i<n<<1;i++) {
			cin>>opt[i].x>>opt[i].y;
			opt[i].flag=1;
		}
		sort(opt,opt+(n<<1));
		printf("%.3f\n",solve(0,(n<<1)-1));
	}
	return 0;
}