题目大意:给定一个无向图,如果它没有环且是连通的则证明是树,那么就要找出一个结点集,使得集合中的任意一点作为树的根节点,都能使得树的深度达到最大。否则,如果该图不是树,则要求输出图的连通分量的个数。
思路:本题可以用并查集或者dfs求连通分量个数。假设已经是树了,如何求deepest root呢?很显然的一个做法就是对树中的每个结点进行dfs搜索,每一次dfs都记下树能达到的最大高度,则在此过程中使得树的深度最大的所有root结点即为所求,但这样暴力求解会有一个case超时。事实上,可以任选一个结点作为起点进行dfs搜索,找到距离起点最远的结点集合A,然后再以A中的任一结点进行dfs,找到距离该结点最远的结点集合B,则A∪B即为所求。(两次dfs的目的就是求树的直径)
代码如下:
#include <iostream>
#include <set>
#include <cstdio>
using namespace std;
const int MAXV = 10005;
int n;
int G[MAXV][MAXV] = {0};
bool vis[MAXV] = {false};
bool flag[MAXV] = {false};
int max_depth = 0;
set<int> deepestRoot;
void dfs(int u, int depth)
{
if(depth > max_depth)
{
max_depth = depth;
deepestRoot.clear();
deepestRoot.insert(u);
}
if(depth == max_depth)
deepestRoot.insert(u);
for(int i = 1; i <= n; i++)
{
if(G[u][i] && !vis[i])
{
vis[i] = true;
dfs(i, depth + 1);
}
}
}
int main()
{
//freopen("in_1021.txt", "r", stdin);
cin >> n;
int start, end;
for(int i = 0; i < n - 1; i++)
{
cin >> start >> end;
G[start][end] = G[end][start] = 1;
}
int k = 0;
for(int i = 1; i <= n; i++)
if(!vis[i]) dfs(i, 1), k++;
if(k > 1)
cout <<"Error: " << k << " components" << endl;
else
{
vis[1] = true;
dfs(1,1);
set<int> root = deepestRoot;
set<int>::iterator p = deepestRoot.begin();
fill(vis, vis + MAXV, 0);
max_depth = 0; //不知道为什么一定要重置max_depth
vis[*p] = true;
dfs(*p, 1);
set<int>::iterator q = root.begin();
for(; q != root.end(); q++)
deepestRoot.insert(*q);
p = deepestRoot.begin();
for(; p != deepestRoot.end(); p++)
cout << *p << endl;
}
}