奶酪

emmmm 一道并查集的题
计算每个球是否相切或者相交 如果是的话归为一个并查集
然后单独记录一下与上表面相切或者相交的球
以及与下表面相切或者相交的球

#include <cstdio>
#include <iostream>
#include <cstring>
#define int long long 
using namespace std;
const int maxn=1000001;
int t,n,h,r,f[maxn],top[maxn],bot[maxn];
int x[maxn],y[maxn],z[maxn];

inline int dis(int i,int j)
{
	return(x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j])+(z[i]-z[j])*(z[i]-z[j]);
}

int find(int x)
{
	if(f[x]!=x)	f[x]=find(f[x]);
	return f[x];		
}

signed main()
{
	scanf("%lld",&t);
	while(t--)
	{
		scanf("%lld %lld %lld",&n,&h,&r);
		int down=0,head=0;
		for(int i=1;i<=n;i++)
			f[i]=i;
		for(int i=1;i<=n;i++)
		{
			scanf("%lld %lld %lld",&x[i],&y[i],&z[i]);
			if(z[i]+r>=h)
				top[++head]=i;
			if(z[i]-r<=0)
				bot[++down]=i;
			for(int j=1;j<=i;j++)
			{
				int disij;
				disij=dis(i,j);
				if(disij>4*r*r)	continue;
				int fi=find(i),fj=find(j);
				if(fi!=fj)
					f[fj]=fi;
			}
		}
		int flag=0;
		for(int i=1;i<=down;i++)
		{
			for(int j=1;j<=head;j++)
			{
				if(find(top[j])==find(bot[i]))
				{
					flag=1;
					break;
				}
			}
			if(flag==1)
				break;
		}
		if(flag==1)	printf("Yes\n");
		else printf("No\n");
	}
	return 0;
}