1336:【例3-1】找樹根和孩子
阿新 • • 發佈:2021-06-27
【題目描述】
給定一棵樹,輸出樹的根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; }