奶酪

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld
题目描述
现有一块大奶酪，它的高度为 h，它的长度和宽度我们可以认为是无限大的，奶酪中间有许多半径相同的球形空洞。我们可以在这块奶酪中建立空间坐标系， 在坐标系中，奶酪的下表面为 z = 0，奶酪的上表面为 z = h。
现在， 奶酪的下表面有一只小老鼠 Jerry， 它知道奶酪中所有空洞的球心所在的坐标。如果两个空洞相切或是相交，则 Jerry 可以从其中一个空洞跑到另一个空洞，特别地，如果一个空洞与下表面相切或是相交， Jerry 则可以从奶酪下表面跑进空洞； 如果一个空洞与上表面相切或是相交， Jerry 则可以从空洞跑到奶酪上表面。
位于奶酪下表面的 Jerry 想知道， 在不破坏奶酪的情况下，能否利用已有的空洞跑到奶酪的上表面去?
​
输入描述:
每个输入文件包含多组数据。
输入文件的第一行，包含一个正整数 T，代表该输入文件中所含的数据组数。
接下来是 T 组数据，每组数据的格式如下：
第一行包含三个正整数 n， h 和 r， 两个数之间以一个空格分开，分别代表奶酪中空洞的数量，奶酪的高度和空洞的半径。
接下来的 n 行，每行包含三个整数 x, y, z， 两个数之间以一个空格分开， 表示空洞球心坐标为 (x,y,z)。
输出描述:
输出文件包含 T 行，分别对应 T 组数据的答案，如果在第 i 组数据中， Jerry 能从下表面跑到上表面，则输出“Yes”，如果不能，则输出“No”（均不包含引号）。

输出：

3
2 4 1
0 0 1
0 0 3
2 5 1
0 0 1
0 0 4
2 5 2
0 0 2
2 0 4

输入：

Yes
No
Yes
思路:做法一：并查集
做法二：搜索（略，暴力，贴个代码）

并查集，如果两个洞相连就放在一个集合，最后只要判断第0和n+1个洞（即最下面和最上面）是否在同一集合。

//并查集
#include <bits/stdc++.h>

#define maxn 0x3f3f3f3f
using namespace std;

typedef long long ll;

int n,m;
ll r,h;
int fa[10010];
int find(int x)
{
    return fa[x]==x ? x:fa[x]=find(fa[x]);
}
void merge(int x,int y)
{
    fa[find(x)]=find(y);
}
struct dong
{
    ll x,y,z;
}t[1005];

bool xianglian(int i,int j) //判断两个洞是否联通
{
    if((t[i].x-t[j].x)*(t[i].x-t[j].x)
       +(t[i].y-t[j].y)*(t[i].y-t[j].y)
       +(t[i].z-t[j].z)*(t[i].z-t[j].z)
       <= (4*r*r))
    return true;
    else return false;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    int T;
    cin>>T;
    while(T--)
    {
        memset(t,0,sizeof(t));
        memset(fa,0,sizeof(fa));
         cin>>n>>h>>r;
        for(int i=0;i<=n+1;i++) // 集合根初始化
            fa[i]=i;
        
       
        for(int i=1;i<=n;i++)
        {
            cin>>t[i].x>>t[i].y>>t[i].z;
            if(t[i].z <= r) merge(0,i);
            if(h-t[i].z <= r) merge(i,n+1);
        }
        for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            if( i!=j && xianglian(i,j) )    merge(i,j);

        if(find(0)==find(n+1) ) cout<<"Yes"<<endl;
        else cout<<"No"<<endl;
    }
}

//搜索
#include <bits/stdc++.h>

#define maxn 0x3f3f3f3f
using namespace std;

typedef long long ll;

int n,m;
ll r,h;
int v[1005];    //洞是否访问
int mp[1005][1005]; //两个点空洞是否相连
struct dong
{
    ll x,y,z;
}t[1005];

int dfs(int x, int y) //搜索，能否和最后一个点联通
{
    if(x==y)    return 1;
    v[x]=1;
    for(int i=1;i<=n+1 ;i++)
    {
        if( !v[i] && mp[x][i])
        {
            if(dfs(i,n+1))  return 1;
        }
    }
    return 0;
}
bool xianglian(int i,int j) //判断两个洞是否联通
{
    if((t[i].x-t[j].x)*(t[i].x-t[j].x)
       +(t[i].y-t[j].y)*(t[i].y-t[j].y)
       +(t[i].z-t[j].z)*(t[i].z-t[j].z)
       <= 4*r*r)
    return true;
    else return false;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);

    int T;
    cin>>T;
    while(T--)
    {
        memset(v,0,sizeof(v));
        memset(mp,0,sizeof(mp));
        cin>>n>>h>>r;
        for(int i=1;i<=n;i++)
        {
            cin>>t[i].x>>t[i].y>>t[i].z;
            if(t[i].z <= r) mp[0][i]=1;
            if(h-t[i].z <= r) mp[i][n+1]=1;
        }
        for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            if( i!=j && xianglian(i,j) )    mp[i][j]=1;

        if(dfs(0,n+1) ) cout<<"Yes"<<endl;
        else cout<<"No"<<endl;
    }
}