P3958 [NOIP2017 提高组] 奶酪

知识点：深度优先搜索

难度：3

这个题就是连通性的判断，每个洞就是一个元素，让我们看有没有一个连通块同时满足有和底面接触的元素，有和顶面接触的元素，这里使用的深度优先搜索的方法，广搜和并查集也都可以，但是并查集还没有学，就不用了，深搜就是先建图，然后搜索就完事了，但是一开始我把高度当成y，并且上了个厕所，结果35分钟才过了这个题，

#include <bits/stdc++.h>

using namespace std;

typedef long long LL;

const int N = 1005;

struct node {
	LL x, y, z;
} a[N];

int n, vis[N];
int tot, ver[N * N * 2], nxt[N * N * 2], head[N];
LL h, r;
bool flag;

LL calc(LL x1, LL y1, LL z1, LL x2, LL y2, LL z2) {
	return (x1 - x2) * (x1 - x2) + (y1 - y2) * (y1 - y2) + (z1 - z2) * (z1 - z2);
}

void add(int x, int y) {
	ver[++tot] = y;
	nxt[tot] = head[x]; head[x] = tot;
}

void dfs(int x) {
	vis[x] = 1;
	if (a[x].z + r >= h) flag = true;
	for (int i = head[x]; i; i = nxt[i]) {
		int y = ver[i];
		if (!vis[y]) dfs(y);
	}
}

int main() {
	int T;
	cin >> T;
	while (T--) {
		cin >> n >> h >> r;
		tot = 0;
		memset(head, 0, sizeof(head));
		for (int i = 0; i < n; i++) {
			cin >> a[i].x >> a[i].y >> a[i].z;
			for (int j = 0; j < i; j++) {
				if (calc(a[i].x, a[i].y, a[i].z, a[j].x, a[j].y, a[j].z) <= 4 * r * r) {
					add(i, j); add(j, i);
				}
			}
		}
		memset(vis, 0, sizeof(vis));
		flag = false;
		for (int i = 0; i < n; i++) {
			if (!vis[i] && a[i].z <= r) {
				dfs(i);
			}
		}
		cout << (flag ? "Yes" : "No") << endl;
	}
	return 0;
}