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

题目连接

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

题目描述

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

输入

第一行：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

解答思路

此题考查的主要是树的内容 但细心观察此题目发现它并不是一个节点只能有1个或两个孩子 而是有多个孩子 所以我们要换种思考方式

我们知道STL中有map容器 有对应的key和value 其中的value我们使用vector容器就能解决一个多个孩子的问题

但新的问题随之而来 如何寻找到此节点的其父节点呢?

所以还是需要一个数组进行父节点标记

#include <bits/stdc++.h>
using namespace std;
int main(){
    int n,m,a,b;
    cin>>n>>m;
    map<int,vector<int>> tree;//构建树结构
    vector<int> parent(n+1,-1);//进行初始化所有节点值为-1 用于标记父节点
    for(int i=0;i<m;i++){
        cin>>a>>b;
        tree[a].push_back(b);
        parent[b] = a;
    }
    int max=0,maxnode=0,root=-1;
    for(int i=1;i<=tree.size();i++){
        if(parent[i]==-1)//查找父节点
            root=i;
        if(tree[i].size()>maxnode){//查找最大结点max
            maxnode = tree[i].size();
            max = i;
        }
    }
    cout<<root<<endl;
    cout<<max<<endl;
    for(auto a:tree[max]){
        cout<<a<<" ";
    }

}