1336：【例3-1】找树根和孩子

【题目描述】

给定一棵树，输出树的根rootroot，孩子最多的结点maxmax以及他的孩子。

【输入】

第一行：nn（结点个数≤100≤100），mm（边数≤200≤200）。

以下mm行：每行两个结点xx和yy，表示yy是xx的孩子(x,y≤1000x,y≤1000)。

【输出】

第一行：树根：rootroot；

第二行：孩子最多的结点maxmax；

第三行：maxmax的孩子（按编号由小到输出）。

【输入样例】

8 7
4 1
4 2
1 3
1 5
2 6
2 7
2 8

【输出样例】

4
2 
6 7 8

#include <bits/stdc++.h>
using namespace std;

int main()
{
    int n, m;
    cin >> n >> m;
    vector<int> dad(n + 1);
    vector <set<int>> son(n + 1);
    int x, y; // y是x的孩子
    for (int i = 0; i < m; i++) {
        cin >> x >> y;
        dad[y] = x;
        son[x].insert(y);
    }
    int root = x;
    while (dad[root] != 0) {
        root = dad[root];
    }
    int max = 1;
    for (int i = 2; i < son.size(); i++) {
        if (son[max].size() < son[i].size()) {
            max = i;
        }
    }
    cout << root << endl;
    cout << max << endl;
    for (set<int>::iterator it = son[max].begin();
         it != son[max].end(); ++it) {
        cout << *it << " ";
    }
    return 0;
}