PAT 1021 Deepest Root (25 分)(dfs)

最新推荐文章于 2022-03-06 17:13:01 发布

兔蠢蠢

最新推荐文章于 2022-03-06 17:13:01 发布

阅读量171

点赞数

分类专栏： PAT 文章标签： PAT 图

本文链接：https://blog.youkuaiyun.com/ftx456789/article/details/93342420

版权

PAT 专栏收录该内容

12 篇文章

订阅专栏

博客围绕一个有n个点n - 1条边的图展开。先判断图的连通性，若连通分量不为1则图不连通，输出连通分量数；若为1，则利用dfs求每个点所能达到的最远距离，找出深度最大时作为根的点，若答案不唯一则按升序输出。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意

给你一个有n个点n-1条边的图，若让某一个点作为根会获得一颗树，求深度最大时是以哪一个点作为根，若答案不唯一按升序输出点，若图不连通或者不能构成树输出其有几个联通分量

思路

因为是一个n个点n-1条边的图所以遍历一遍图可以得到图的连通分量，如果连通分量不为1则图肯定不连通不能构成一棵树，输出其联通分量。若联通分量为0，则我们利用dfs求每一个点所能达到的最远距离，这个时候需要一个全局变量来记录dfs过程中以x为根所达到的距离

#include<bits/stdc++.h>
using namespace std;
const int N=1e4+5;
vector<int>E[N];
bool vis[N];
int d[N];
int maxn=0;
int dfs(int x,int step)
{
    for(int i=0; i<E[x].size(); i++)
    {
        if(!vis[E[x][i]])
        {
            vis[E[x][i]]=1;
            maxn=max(maxn,dfs(E[x][i],step+1));
            vis[E[x][i]]=0;
        }
    }
    return step;
}
void dfs2(int x)
{
    vis[x]=1;
    for(int i=0; i<E[x].size(); i++)
    {
        if(!vis[E[x][i]])
            dfs2(E[x][i]);
    }
}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=0; i<n-1; i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        E[u].push_back(v);
        E[v].push_back(u);
    }
    int cnt=0;
    for(int i=1; i<=n; i++)
        if(!vis[i])
        {
            cnt++;
            dfs2(i);
        }
    if(cnt!=1)
    {
        printf("Error: %d components\n",cnt);
        return 0;
    }
    for(int i=1;i<=n;i++)
        vis[i]=0;
    int deep=0;
    for(int i=1; i<=n; i++)
    {
        vis[i]=1;
        maxn=0;
        dfs(i,1);
        d[i]=maxn;
        vis[i]=0;
        deep=max(deep,d[i]);
    }
    for(int i=1;i<=n;i++)
        if(deep==d[i])
        printf("%d\n",i);
    return 0;
}