PAT 1021. Deepest Root

【题目链接】

题目大意：给定一个无向图，如果它没有环且是连通的则证明是树，那么就要找出一个结点集，使得集合中的任意一点作为树的根节点，都能使得树的深度达到最大。否则，如果该图不是树，则要求输出图的连通分量的个数。

思路：本题可以用并查集或者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;
    }
}