PTA1021 Deepest Root (25分)

题目：click me~

题意：给n个点，n-1条边，判断是否连通图。

若是，找出能使树高最高的根结点（如果这样的结点不止一个，那么升序输出它们）；

若不是，输出连通块个数。

解题思路：

步骤一：判断是否连通图；

步骤二：先任意选择一个结点开始遍历树，获取能达到的最深的顶点（记为A集合）。若是连通图，再从A中任选择一个结点出发遍历树，获取能达到的最深的顶点（记为B集合），那么A、B的并集即为所求的deepest root。

Tips：

1.标志一个visit数组，每次dfs时，都把访问到的结点visit【i】置为1，用循环遍历所有结点，没有visit的结点再进行dfs，连通块个数++。

2.STL中的set自动排序、去重，可以直接存放deepest root。

code：

#include <iostream>
#include <vector>
#include <set>
#include <algorithm>
using namespace std;
int n, maxheight = 0;
vector<vector<int>> v;
bool visit[10010];
set<int> s;
vector<int> temp;
void dfs(int node, int height) {
    if(height > maxheight) {
        temp.clear();
        temp.push_back(node);
        maxheight = height;
    } else if(height == maxheight){
        temp.push_back(node);
    }
    visit[node] = true;
    for(int i = 0; i < v[node].size(); i++) {
        if(visit[v[node][i]] == false)
            dfs(v[node][i], height + 1);
    }
}
int main() {
    scanf("%d", &n);
    v.resize(n + 1);
    int a, b, cnt = 0, s1 = 0;
    for(int i = 0; i < n - 1; i++) {
        scanf("%d%d", &a, &b);
        v[a].push_back(b);
        v[b].push_back(a);
    }
    for(int i = 1; i <= n; i++) {
        if(visit[i] == false) {
            dfs(i, 1);
            if(i == 1) {
                if (temp.size() != 0) s1 = temp[0];
                for(int j = 0; j < temp.size(); j++)
                    s.insert(temp[j]);
            }
            cnt++;
        }
    }
    if(cnt >= 2) {
        printf("Error: %d components", cnt);
    } else {
        temp.clear();
        maxheight = 0;
        fill(visit, visit + 10010, false);
        dfs(s1, 1);
        for(int i = 0; i < temp.size(); i++)
            s.insert(temp[i]);
        for(auto it = s.begin(); it != s.end(); it++)
            printf("%d\n", *it);
    }
    return 0;
}