算法实验T7——POJ3714 Raid

题目链接：

题目大意：

有两组坐标，一组是 n 个发电站的坐标，一组是 n 个士兵的坐标。

求哪个士兵离发电站的距离最短。

思路：

        联想到求解二维两点最短距离这样一个经典的二分问题：我们从最终找到的相距最短的两点的分布入手，先将所有点按 x 坐标排序后二分为左右两部分，那么这两点要么都在左边、要么都在右边，要么一左一右。如果都在同侧的话，这就是原问题的子问题。由此，我们先二分求解左右两边子问题的最优解，然后讨论一左一右的情况。这里我们已经得到了一个解mi（两子问题中更优的解），那么可以根据这个解来剪枝：由于我们已经按x坐标排序，那么对于那些到最中间点的x方向距离（即）已经大于mi的点，其和另外一部分中任何一个点的距离一定大于mi，无需再考虑。然后再将筛选出的点按 y 坐标排序，同理剪枝；然后在这些点中，两两求距离更新最值。

        而题中只有一点不同，就是点分为了两种类型，求的是一种类型（士兵）到另一种类型（发电站）最小距离，不能直接套用上述解法。其实只需要将相同类型的点之间的距离预设为无穷大，这样就可以消除类型差异的影响，把两种点当成一种来处理。

AC代码：

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<iostream>
#include<cstdio>
using namespace std;
const double INF = 1e48;
typedef struct {
	double x, y;
	int type;
}NODES;
NODES nodes[200005], temp[200005];
bool cmpx(NODES a, NODES b) {
	return a.x < b.x;
}
bool cmpy(NODES a, NODES b) {
	return a.y < b.y;
}
double pw(double x) {
	return x * x;
}
double dis(NODES a, NODES b) {
	if (a.type == b.type) {
		return INF;
	}//同为士兵或者同为发电站则距离设为无穷大，这样就可以把两者放在一起讨论。
	else return sqrt((a.x - b.x)*(a.x - b.x) + (a.y - b.y)*(a.y - b.y));
}
double mindis(long long int l, long long int r) {
	if (l == r) {
		return INF;
	}
	if (r - l == 1) {
		return dis(nodes[l], nodes[r]);
	}
	long long int mid = (l + r) / 2;
	double mi = INF, mil, mir;
	long long  cnt = 0;
	mil = mindis(l, mid);//取得最短距离的两点都在左边的情况
	mir = mindis(mid + 1, r);//取得最短距离的两点都在右边的情况
	mi = min(mil, mir);
	for (int i = l; i <= r; i++) {
		if (abs(nodes[i].x - nodes[mid].x) <= mi) {
			temp[cnt++] = nodes[i];
		}
	}//如果到中间的横向距离已经大于目前最小则不用再考虑。
	sort(temp, temp + cnt, cmpy);
	for (int i = 0; i < cnt; i++) {
		for (int j = i + 1; j < cnt; j++) {
			if (abs(temp[i].y - temp[j].y) > mi) {//同理，如果到中间的纵向距离已经大于目前最小则不用再考虑。
				break;
			}
			mi = min(mi, dis(temp[i], temp[j]));
		}
	}
	return mi;
}

int main() {
	int T;
	long long N;
	const int STATION = 0, AGENT = 1;
	cin >> T;
	for (int i = 0; i < T; i++) {
		cin >> N;
		for (int j = 0; j < N; j++) {
			cin >> nodes[j].x >> nodes[j].y;
			nodes[j].type = STATION;
		}
		for (int j = N; j < 2 * N; j++) {
			cin >> nodes[j].x >> nodes[j].y;
			nodes[j].type = AGENT;
		}
		sort(nodes, nodes + 2 * N, cmpx);
		printf("%.3lf\n", mindis(0, 2 * N - 1));
	}

	return 0;
}