洛谷 P3958 奶酪

原题链接

这个题并查集比较巧妙的一点就是，将上表面和下表面看做n+1和n，从0到n连接各个点， 如果洞和上表面相交或者相切，则连接，如果和下表面相切或者相交，那么连接，最后检查上表面和下表面是不是同一个祖先即可。

并查集做法：

#include <iostream>
#include<queue>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <cmath>
typedef long long ll;
using namespace std;
const int maxn = 100000 + 10;
ll t,fa[1001],flag1=0,flag2=0;
struct node{
	ll x,y,z;
}a[1001];
inline ll find(ll x){return x==fa[x]?x:fa[x]=find(fa[x]);}
inline void Union(ll x,ll y){
	ll px=find(x);
	ll py=find(y);
	if(px!=py) fa[px]=py;
}
inline double dist(double x,double y,double z,double aa,double ba,double ca)
{
	return pow((x-aa),2)+pow((y-ba),2)+pow((z-ca),2);
}
int main(){
	scanf("%lld",&t);
	while(t--){
		flag1=0,flag2=0;
		ll n,h,r;
		scanf("%lld%lld%lld",&n,&h,&r);
		for(ll i=0;i<=n+1;i++) fa[i]=i;
		for(ll i=0;i<n;i++) scanf("%lld %lld %lld",&a[i].x,&a[i].y,&a[i].z);
		for(ll i=0;i<n;i++){
			for(ll j=i+1;j<n;j++){
				if(dist(a[i].x,a[i].y,a[i].z,a[j].x,a[j].y,a[j].z)<=4*r*r)
					Union(i,j);
			}
			if(a[i].z<=r) Union(i,n);
			if(a[i].z+r>=h) Union(i,n+1);
		}
		if(find(n)!=find(n+1)) printf("No\n");
		else printf("Yes\n");
	}
	return 0;
}

深度优先搜索做法：

思路就是，找到一个与下表面相交或者相切的点，然后开始从这个点开始遍历各个点，如果出现了与上表面相切或者相交的点，那么就输出Yes,否则输出No。

#include <iostream>
#include<queue>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <cmath>
#define mm(a,b) memset(a,b,sizeof(a))
typedef long long ll;
using namespace std;
const int maxn = 100000 + 10;
ll t,fa[1001],flag=0;
ll n,h,r;
ll vis[1001];
struct node{
	ll x,y,z;
}a[1001];
inline ll dist(ll x,ll y,ll z,ll x1,ll y1,ll z1){
	return (x-x1)*(x-x1)+(y-y1)*(y-y1)+(z-z1)*(z-z1);
}
void dfs(int k){
	if(flag==1) return;
	if(a[k].z+r>=h){
		flag=1;
		return ;
	}
	for(int i=0;i<n;i++){
		if(!vis[i]){
			ll t=dist(a[i].x,a[i].y,a[i].z,a[k].x,a[k].y,a[k].z);
			if(t<=4*r*r){
				vis[i]=1;
				dfs(i);
			}
		}
	}
} 
int main(){
	scanf("%lld",&t);
	while(t--){
		flag=0;
		mm(vis,0);
		scanf("%lld%lld%lld",&n,&h,&r);
		for(int i=0;i<n;i++) cin>>a[i].x>>a[i].y>>a[i].z;
		for(int i=0;i<n;i++){
			if(a[i].z<=r){
				vis[i]=1;
				dfs(i);
			}
		}
		if(flag==1) printf("Yes\n");
		else printf("No\n");
	}
	return 0;
}

最后的最后一定要注意精度问题！