1021 Deepest Root（树的直径、bfs/dfs、并查集）

 先通过并查集判断有几个连通图，如果只有一张图，那就用两次dfs/bfs来找到树的直径上的所有端点

#include<bits/stdc++.h>
using namespace std;
int n;
vector<int> edges[10005];
bool visited[10005] = {false};
set<int> temp; //记录该次dfs筛选出树直径上的端点
set<int> res; //所有树直径上的端点
vector<int> parents(10005); //父节点，用于并查集
int maximum = -1;
int find(int x) {
    if (x == parents[x]) {
        return x;
    }
    parents[x] = find(parents[x]); //路径压缩
    return parents[x];
}
void merge(int a, int b) {
    int pa = find(a);
    int pb = find(b);
    if (pa != pb) {
        parents[pb] = pa;
    }
}

void dfs(int p, int level) {
    if (level > maximum) {
        temp.clear();
        temp.insert(p);
        maximum = level;
    } else if (level == maximum) {
        temp.insert(p);
    }
    visited[p] = true;
    for (auto i : edges[p]) {
        if (!visited[i]) {
            dfs(i, level + 1);
        }
    }
}
int main() {
    cin >> n;
    int a, b;
    for (int i = 1; i <= n; ++ i) {
        parents[i] = i;
    }
    for (int i = 0; i < n - 1; ++ i) {
        cin >> a >> b;
        edges[a].push_back(b);
        edges[b].push_back(a);
        merge(a,b);
    }
    int components = 0;
    for (int i = 1; i <= n; ++ i) {
        if (find(i) == i) {
            components ++;
        }
    }
    if (components > 1) {
        printf("Error: %d components", components);
        return 0;
    }
    int root = 1;
    dfs(root, 0); //第一次dfs用于找到树直径上的端点
    for (auto i : temp) {
        res.insert(i);
        root = i; //取任意一个树直径上的端点
    }
    memset(visited, false, sizeof(visited));
    temp.clear();
    dfs(root, 0); //以树直径上的端点为根进行遍历，得到所有树直径上的端点
    for (auto i : temp) {
        res.insert(i);
    }
    for (auto i : res) {
        cout << i << endl;
    }

}