1. 程式人生 > 其它 >1336:【例3-1】找樹根和孩子

1336:【例3-1】找樹根和孩子

【題目描述】

給定一棵樹,輸出樹的根rootroot,孩子最多的結點maxmax以及他的孩子。

【輸入】

第一行:nn(結點個數100≤100),mm(邊數200≤200)。

以下mm行:每行兩個結點xx和yy,表示yy是xx的孩子(x,y1000x,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; }