P3958 [NOIP2017 提高组] 奶酪

题目链接：https://www.luogu.com.cn/problem/P3958

题目描述

现有一块大奶酪，它的高度为
h，它的长度和宽度我们可以认为是无限大的，奶酪中间有许多半径相同的球形空洞。我们可以在这块奶酪中建立空间坐标系，在坐标系中，奶酪的下表面为
z=0，奶酪的上表面为 z=h。

现在，奶酪的下表面有一只小老鼠 Jerry，它知道奶酪中所有空洞的球心所在的坐标。如果两个空洞相切或是相交，则 Jerry
可以从其中一个空洞跑到另一个空洞，特别地，如果一个空洞与下表面相切或是相交，Jerry
则可以从奶酪下表面跑进空洞；如果一个空洞与上表面相切或是相交，Jerry 则可以从空洞跑到奶酪上表面。

位于奶酪下表面的 Jerry 想知道，在不破坏奶酪的情况下，能否利用已有的空洞跑 到奶酪的上表面去?

空间内两点
P1(x1,y1,z1)、P2(x2,y2,z2) 的距离公式如下：

dist(P1,P2) = sqrt ( (x1−x2)²+(y1−y2)²+(z1−z2)² )

​ 输入格式

每个输入文件包含多组数据。

第一行，包含一个正整数 T，代表该输入文件中所含的数据组数。

接下来是 T 组数据，每组数据的格式如下：
第一行包含三个正整数 n , h , r 两个数之间以一个空格分开，分别代表奶酪中空洞的数量，奶酪的高度和空洞的半径。

接下来的 n行，每行包含三个整数 x,y,z，两个数之间以一个空格分开，表示空洞球心坐标为
(x,y,z)。
输出格式

T行，分别对应 T组数据的答案，如果在第 i 组数据中，Jerry 能从下表面跑到上表面，则输出 Yes，如果不能，则输出No。
输入输出样例
输入 #1

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

输出 #1

Yes
No
Yes

思路：
如果能想到用并查集解决的话那么思路就很简单了。首先选定0，n+1作为底和顶，对输入的球进行合并，如果最终底部和顶部在同一个集合（有相同祖先）那么说明Jerry可以跑到上表面。
简单的并查集应用~

#include<iostream>
#include<cstring>
typedef long long LL;
using namespace std;
const int maxn = 1e5 + 5;
const int s = 1e3 + 5;
int fa[s];
//int deep[s];
LL x[s], y[s], z[s];
int t, n;
int h, r;
void init() {
	for (int i = 0; i <= n + 1; i++) fa[i] = i;//, deep[i] = 1;
}
bool isTangency(LL x1, LL y1, LL z1, LL x2, LL y2, LL z2, LL r) {
	return{ (x1 - x2)*(x1 - x2) + (y1 - y2)*(y1 - y2) + (z1 - z2)*(z1 - z2) <= 4 * r*r };
}
int find(int x) {
	//if (fa[x] == x) return x;
	//return find(fa[x]);
	//return (fa[x] = find(fa[x]));
	//路径压缩
	//{
	 return x==fa[x]?x:(fa[x]=find(fa[x]));
    //}
}
void merge(int x, int y) {
	//********普通合并
	fa[find(x)] = find(y);
	
	//*******按深度合并(与路径压缩一同使用，会导致deep[]深度异常)但貌似影响不大
	//int fx = find(x), fy = find(y);//先找到两个节点的根节点

	//if (deep[fx] <= deep[fy]) {//如果后者深度更深则以后者作为根节点
	//	fa[fx] = fy;
	//}
	//else fa[fy] = fx;
	//if (deep[fx] == deep[fy] && fx != fy) {//如果两个节点深度相同且没有公共祖先
	//	deep[fy] += 1;
	//}

}
int main() {
	cin >> t;
	while (t--) {
		memset(x, 0, sizeof(x));
		memset(y, 0, sizeof(y));
		memset(z, 0, sizeof(z));
		cin >> n >> h >> r;
		init();
		for (int i = 1; i <= n; i++) {
			cin >> x[i] >> y[i] >> z[i];
		}
		for (int i = 1; i <= n; i++) {
			if (h - z[i] <= r) merge(i, n + 1);//将与顶部相接触的洞与顶部合并
			if (z[i] <= r) merge(i, 0);//与底部合并
		}
		for (int i = 1; i <= n; i++) {
			for (int j = i + 1; j <= n; j++) {
				if (isTangency(x[i], y[i], z[i], x[j], y[j], z[j], r)) {
					merge(i, j);
				}
			}
		}
		//cout << find(0) << ' ' << find(n + 1) << '\n';
		if (find(0) == find(n + 1)) cout << "Yes\n";
		else cout << "No\n";
	}
	return 0;
}