pat 甲级 1021 Deepest Root

中文意思：

一个无环连通图可以被视作一个树。

树的高度取决于所选取的根节点。

现在，你要找到可以使得树的高度最大的根节点。

它被称为最深的根。

输入格式

第一行包含整数 N，表示节点数量。

节点编号为 1∼N。

接下来 N−1 行，每行包含两个整数，表示两个节点之间存在一条边。

输出格式

输出最深的根的节点编号。

如果最深的根不唯一，则按照从小到大的顺序，将它们依次输出，每个占一行。

如果给定的图不是树，输出 Error: K components，其中 K是图中连通分量的数量。

----------------------------------------------------------------------------------------------------------------

解题思路：（1）首先是如何判断给定的图不是树。（2）如何判断最深的根的节点编号。

1.如何判断给定的图不是树，因为题目要输出图中连通分量的数量，我们可以想到用并查集的思想，即一开始每个节点都是一个连通分量，当节点有边时，合并成一个集合。这思想与pat的1013题相似，具体细节跳转链接（pat 甲级 1013 Battle Over Cities-CSDN博客）。

2.如何判断最深的根的节点。我们可以用邻接表来存储树，然后dfs递归遍历。这思想与pat的1004题相似，具体细节跳转（pat 甲级 1004 Counting Leaves-CSDN博客）。

整体代码如下：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
const int N=10010,M=N*2;
int p[N];//并查集使用的数组，表示i节点的祖宗节点
int n;
int h[N],e[M],ne[M];//邻接表存储树所需的数组
int idx;
int find(int x){//压缩路径，原理看查看上文的链接
    if(p[x]!=x) p[x]=find(p[x]);
    return p[x];
}
void add(int a,int b){//邻接表存储边
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
int dfs(int u,int father){//第一个变量表示从u开始dfs，第二个变量表示父节点，因为是无向图，标记 防止搜回父节点。
    int depth=0;
    for(int i=h[u];~i;i=ne[i]){//遍历邻接表
        int j=e[i];
        if(j==father) continue;
        depth=max(depth,dfs(j,u)+1);
    }
    return depth;
}
int main()
{
    cin>>n;
    memset(h,-1,sizeof h);//邻接表的头节点全初始化为-1
    for(int i=1;i<=n;i++) p[i]=i;//并查集初始化
    int k=n;//n个节点，n个连通分量
    for(int i=0;i<n-1;i++){
        int a,b;
        cin>>a>>b;
        int pa=find(a),pb=find(b);//查找祖宗节点
        if(pa!=pb){//不在同一个集合，有边则合并，连通分量-1
            p[pa]=pb;
            k--;
        }
        add(a,b);//添加边，无向图相当于a->b,b->a的有向图，添加两条边
        add(b,a);
    }
    if(k>1) printf("Error: %d components\n",k);
    else{
        vector<int> nodes;//存储最深的nodes
        int max_depth=-1;
        for(int i=1;i<=n;i++){
            int depth=dfs(i,-1);//第一个变量表示从i开始dfs，第二个变量表示父节点，因为是无向图，标记 防止搜回父节点。
            if(depth>max_depth){//有更深的结果
                nodes.clear();//清理所有的nodes
                nodes.push_back(i);//将最新的节点放入
                max_depth=depth;//更新depth
            }
            else if(depth==max_depth){
                nodes.push_back(i);
            }
        }
        for(auto v:nodes){
            cout<<v<<endl;
        }
    }
    return 0;
}