图论专题--P3958 [NOIP2017 提高组] 奶酪

 

 可以用并查集来做这道题，一棵棵树里存的是一组组联通的空洞，每一棵树存一组联通空洞

首先开了个结构体来具体地表示空洞的坐标x y z ，因为数据范围很大所以必须开longlong

然后用down存储能够直接联通奶酪底部的空洞的数量，用d数组来存这些空洞的序号

每一次访问，，首先要将并查集初始化 ，将down归零

然后依次遍历所有空洞 ，先判断空洞是否直接联通奶酪底部 ，如果是的话存入d数组并down++

然后再将该空洞序号之前的所有空洞遍历一遍，这里find函数返回的是该空洞能够通到的高度最高的空洞的序号，这里把这个高度最高的空洞称为该空洞的老大。

把该空洞之前的所有空洞遍历一遍首先要判断两空洞是否联通，不连通直接跳过就完事，如果联通，要比较一下这个空洞的老大 和当前空洞哪个高度更高，将高度更高的空洞作为树顶加入集合中，也就是让更高的空洞成为这棵树中所有空洞的老大。

把树存完之后，利用之前存的直接联通底部的空洞数组，遍历d数组

依次询问d数组中空洞的老大，如果他的老大的高度足以通向奶酪顶部，那么表示这个直接联通底端的空洞可以通向顶端

#include <bits/stdc++.h>
using namespace std;

struct Point
{
    long long x,y,z;  // 结构体用来存空洞的坐标
}point[1005];

int T,down;     //down代表直接连接底部的空洞数量
long long n,h,r;
int p[1005];    //并查集数组
int d[1005];   //用来存直接连接底部的空洞的序号

int find(int x)  //并查集标准find函数
{
    if(p[x] != x) return find(p[x]);
    else return x;
}

long long dist(Point a, Point b)  //返回两点间距离的平方
{
    return pow(a.x-b.x,2)+pow(a.y-b.y,2)+pow(a.z-b.z,2);
}

int main()
{
    cin >> T;
    for(int k = 1; k <= T; k ++)
    {
        scanf("%lld%lld%lld", &n, &h, &r);  //空洞数量、奶酪高度、半径

        for(int i = 1; i <= n; i ++) p[i] = i;  //并查集初始化

        for(int i = 1; i <= n; i ++)
        {
            scanf("%lld%lld%lld", &point[i].x, &point[i].y, &point[i].z);
            
            if(point[i].z <= r) d[++down] = i; //判断是否直接连接底部
            
            for(int j = 1; j < i; j ++)
            {
                if(i != j && dist(point[i], point[j]) <= 4*r*r && find(i) != find(j))
                {
                    if(point[find(i)].z < point[find(j)].z) p[find(i)] = find(j);
                    else p[find(j)] = find(i);//让高度最高的当老大，树顶存的是是最高的空洞
                }
            }
        }

        bool flag = 0;
        for(int i = 1; i <= down; i ++)
        {  //询问每个底端空洞的老大，看看老大能不能连接顶部
            if(point[find(d[i])].z + r >= h) {flag = 1; break;}
        }
        if(!flag) cout << "No";
        else cout << "Yes";
        if(k!=T) cout << endl;
        down = 0;    //别忘了每次访问前的初始化
    }

    return 0;
}