代码

#include<bits/stdc++.h>
using namespace std;
int t;
int n,h,r;
int x[1001],y[1001],z[1001];
int a[1001];
int s[1001],e[1001];
int ls,le;
int find(int x){
	if(x==a[x]){
		return x;
	}else{
		return (a[x]=find(a[x]));
	}
}
double l(int x1,int y1,int z1,int x2,int y2,int z2){
	return sqrt( pow(x2-x1,2) + pow(y2-y1,2) + pow(z2-z1,2) );
}
void run(){
	for(int i=1;i<=ls;i++){
		for(int j=1;j<=le;j++){
			if(find(s[i])==find(e[j])){
				cout << "Yes\n";
				return;
			}
		}
	}
	cout << "No\n";
	return;
}
int main(){
	cin >> t;
	while(t--){
		ls=le=0;
		cin >> n >> h >> r;
		for(int i=1;i<=n;i++)a[i]=i;
		for(int i=1;i<=n;i++){
			cin >> x[i] >> y[i] >> z[i];
			if(z[i]+r>=h)e[++le]=i;//这个空洞能上去
			if(z[i]-r<=0)s[++ls]=i;//可以作为出发点 
			for(int j=1;j<i;j++){
				if(l(x[i],y[i],z[i],x[j],y[j],z[j]) <= 2*r){
					a[find(i)]=find(j);
				}
			}
		}
	//	for(int i=1;i<=n;i++)cout << a[i] << " ";
		run();
	}
	return 0;
}

评测结果