C++ 奶酪

题目描述

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

输入描述:

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

输出描述:

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

解题思路
这是一道并查集模板题，先找到所有的点，将能够连通的圆圈归为一个集合，然后将所有与顶部和底部相交或相切的点存入不同vector数组，再循环判断是否在一个集合内，如果在一个集合内则这个集合内的洞能够上下连通，输出Yes就行，如果没有就输出No

#include<bits/stdc++.h>
using namespace std;
int pz[1005];
int Rank[1005];
long long  x[1005],y[1005],z[1005];
vector<long long > blow,top;
int n;
long long h,r;
void init(){	//初始化
	for(int i=1;i<=n;i++){
		pz[i]=i;
		Rank[i]=1;
	}
}
int find(int x){	//查找
	return x==pz[x]?x:(find(pz[x]));
}


void merge(long long x1,long long y1,long long z1,int s){
	long long dist;
	for(int i=1;i<=s;i++){
		if((x[i]-x1)*(x[i]-x1)+(y[i]-y1)*(y[i]-y1)+(z[i]-z1)*(z[i]-z1)>4*r*r) continue;
		//cout<<dist<<endl;	//判断两个球形状是不是相交或者相切
		if((x[i]-x1)*(x[i]-x1)+(y[i]-y1)*(y[i]-y1)+(z[i]-z1)*(z[i]-z1)<=4*r*r){
			int l = find(i);
			int p = find(s);
			if(l!=p)pz[l]=p;		//归入集合，这里要将已经输入的集合归于新输入的数据下，因为新输入的球形可能跟好几个点相交，写成pz[p]=l的话会覆盖掉。错了好几次就错在这里
		}
	}
}
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);cout.tie(0);
	int t;
	cin>>t;
	while(t--){
		cin>>n>>h>>r;
		init();
		blow.clear();	//存和上下面相交相切的点
		top.clear();
		for(int i=1;i<=n;i++){
			cin>>x[i]>>y[i]>>z[i];
			merge(x[i],y[i],z[i],i);
			if(z[i]-r<=0){
				blow.push_back(i);
			}
			if(z[i]+r>=h){
				top.push_back(i);
			}
		}
		int sum=0;
		for(int i=0;i<blow.size();i++){
			//cout<<blow[i]<<endl;	//循环判断是否属于一个集合
			for(int j=0;j<top.size();j++){
				//cout<<top[j]<<" "<<find(blow[i])<<" "<<find(top[j])<<endl;
				if(find(blow[i])==find(top[j])){
					sum++;
				}
			}
		}
		if(sum==0){
			cout<<"No"<<endl;
		}else{
			cout<<"Yes"<<endl;
		}
		
	}
}

大佬们有问题帮我指出，谢谢

题目来源
链接：https://ac.nowcoder.com/acm/problem/16417
来源：牛客网

关于并查集有一篇文章写的挺有意思，生动形象
https://blog.csdn.net/qq_41593380/article/details/81146850