【NOIP2017提高组Day 2】奶酪

Description

现有一块大奶酪，它的高度为 h，它的长度和宽度我们可以认为是无限大的，奶酪中间有许多 半径相同 的球形空洞。我们可以在这块奶酪中建立空间坐标系，在坐标系中，奶酪的下表面为z = 0，奶酪的上表面为z = h。
现在，奶酪的下表面有一只小老鼠 Jerry，它知道奶酪中所有空洞的球心所在的坐标。两相切或是相交，则 Jerry 可以从其中一个空洞跑到另一个，特别地，如果一个空洞与下表面相切或是相交，Jerry 则可以从奶酪下表面跑进空洞；如果一个空洞与上表面相切或是相交，Jerry 则可以从空洞跑到奶酪上表面。
位于奶酪下表面的 Jerry 想知道，在 不破坏奶酪 的情况下，能否利用已有的空洞跑到奶酪的上表面去?

Input

每个输入文件包含多组数据。
输入文件的第一行，包含一个正整数 T，代表该输入文件中所含的数据组数。
接下来是 T 组数据，每组数据的格式如下：
第一行包含三个正整数 n，h 和 r，两个数之间以一个空格分开，分别代表奶酪中空洞的数量，奶酪的高度和空洞的半径。
接下来的 n 行，每行包含三个整数 x、y、z，两个数之间以一个空格分开，表示空洞球心坐标为(x,y,z)。

Output

输出文件包含 T 行，分别对应 T 组数据的答案，如果在第 i 组数据中，Jerry 能从下表面跑到上表面，则输出“ Yes ”，如果不能，则输出“ No ”（均不包含引号）。

solution

这道题我的方法是先预处理出那两个点能够互相到达，就是dist(P1,P2)小于2r，两个点就可以互相到达，还要预处理出那些球能够到达底面或者顶面，最后从底面DFS，如果能到达顶面，就输出Yes，否则输出No。

code

#include<bits/stdc++.h>
#define rg register int
using namespace std;
long long T, n, h, r, x[1005], y[1005], z[1005], f[1005][1005], bz[1005], ans;
int dist(int i, int j) {
	double s = sqrt((x[i] - x[j]) * 1.0 * (x[i] -x [j])*1.0 + (y[i] - y[j]) * 1.0 * (y[i] - y[j])*1.0 + (z[i] - z[j]) * 1.0 * (z[i] -z [j]) * 1.0) * 1.0;
	if (s > (int)s) s++;
	return s;
}
void dfs(int x) {
	if (x == n + 1) {
		ans = 1;
		return;
	}
	bz[x] = 1;
	for (rg i = 1; i <= n + 1; i++)
		if (f[x][i] && (!bz[i])) {
			dfs(i);
			if (ans == 1)
				return;
		}
}
int main() {
	scanf("%lld", &T);
	while (T--) {
		scanf("%lld%lld%lld", &n, &h, &r), ans = 0;
		for (rg i = 1; i <= n; i++)
			scanf("%lld%lld%lld", &x[i], &y[i], &z[i]);
		for (rg i = 0; i <= n; i++)
			for(rg j = i + 1; j <= n+1; j++)
				f[i][j] = f[j][i] = 0;
		for (rg i = 1; i <= n; i++)
			for (rg j = i + 1; j <= n; j++)
				if (dist(i,j) <= r * 2) f[i][j] = f[j][i] = 1;
		for (rg i = 1; i <= n; i++) {
			bz[i] = 0;
			if (h - z[i] <= r) f[i][n + 1] = f[n + 1][i] = 1;
			if (z[i] <= r) f[i][0] = f[0][i] = 1;
		}
		dfs(0);
		if (ans) printf("Yes\n");
		else printf("No\n");
	}
	return 0;
}