洛谷P3958 并查集做法详解

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

思路：判断小鼠是否能够逃出，即是判断下方空洞与上方是否连通（也就是是否在一个集合内），一个与地板接壤的空洞，另一个超出奶酪高度的空洞，是连通的话就可以逃出去。

以下是ac代码（附详细注释）

//用并查集解决时，只要判断在满足题意的情况下，低与高是否连通即可。
#include<iostream>
#include<map>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
#define res(i,a,b) for(int i=(a);i<=(b);i++)
#define los(i,a,b) for(int i=(a);i<(b);i++)
#define dwn(i,a,b) for(int i=(a);i>=(b);i--)
#define el '\n'
using LL = long long;
using db = double;

struct point
{
	LL x, y, z;//坐标
}p[1010];
LL r, t, n, h, f[1010];
LL find(int x)
{
	return x == f[x] ? x : f[x] = find(f[x]);//并查集找“老大”的板子
}
bool cmp(point a, point b)
{
	return a.z < b.z;//把空洞的高度从小到大排序，只要判断小的并与地板接壤，大的并超出奶酪高度的两个空洞是否连通，是的话就可以逃出去。
}
LL dist(int a, int b)//返回距离的平方
{
	return (p[a].x - p[b].x) * (p[a].x - p[b].x) + (p[a].y - p[b].y) * (p[a].y - p[b].y) + (p[a].z - p[b].z) * (p[a].z - p[b].z);
}
void solve()
{
	cin >> n >> h >> r;
	res(i, 1, n)
		cin >> p[i].x >> p[i].y >> p[i].z;//读取坐标
	sort(p + 1, p + 1 + n, cmp);//排序
	res(i, 1, n)
		f[i] = i;//并查集初始化
	if (p[1].z > r || p[n].z + r < h)//最矮的都到不了或者最高的都出不去就直接say NO
		cout << "No" << el;
	else
	{
		res(i, 1, n)
		{
			res(j, 1, i)//上限是i，只要判断新的空洞与之前的是否相连就行
			{
				if ((p[i].x - p[j].x) * (p[i].x - p[j].x) + (p[i].y - p[j].y) * (p[i].y - p[j].y) + (p[i].z - p[j].z) * (p[i].z - p[j].z) > 4 * r * r)//防爆LL
					continue;
				if (dist(i, j) <= 4 * r * r)//距离小于两个半径意味着可以相连（这里是同时平方了）
				{
					if (find(i) != find(j))//如果未连通就相连
						f[find(f[i])] = find(f[j]);
				}
			}
		}
		int flag = 1, flag1 = 1;//判断是否能到达，判断循环是否退出
		res(i, 1, n)
		{
			if (p[i].z > r)//我们是排过序的，后面的只会比前面的更高
				break;
			dwn(j, n, i)
			{
				if (p[j].z + r < h)//同样排过序，后面的只会更矮
					break;
				if (find(i) == find(j))//同时满足以上两种情况并且连通，那就可以跑出去了
				{
					flag = 0;
					flag1 = 0;
					break;
				}
			}
			if (!flag1)//说明找到了
				break;
		}
		if (flag)
			cout << "No" << el;
		else
			cout << "Yes" << el;
	}
}
int main()
{
	cin >> t;
	while (t--)
		solve();
}