1004 数叶子结点

家庭关系可以用家谱树来表示,给定一个家谱树,你的任务是找出其中没有孩子的成员。

输入格式

第一行包含一个整数 N 表示树中结点总数以及一个整数 M 表示非叶子结点数。

接下来 M 行,每行的格式为:

ID K ID[1] ID[2] ... ID[K]

ID 是一个两位数字,表示一个非叶子结点编号,K 是一个整数,表示它的子结点数,接下来的 K 个 ID[i] 也是两位数字,表示一个子结点的编号。

为了简单起见,我们将根结点固定设为 01

所有结点的编号即为 01,02,03,,31,32,33,,N

输出格式

输出从根结点开始,自上到下,树的每一层级分别包含多少个叶子节点。

输出占一行,整数之间用空格隔开。

数据范围

0<N<100

输入样例:

2 1
01 1 02

输出样例:

0 1

样例解释

该样例表示一棵只有 2 个结点的树,其中 01结点是根,而 02 结点是其唯一的子节点。

因此,在根这一层级上,存在 0 个叶结点;在下一个级别上,有 1 个叶结点。

所以,我们应该在一行中输出0 1。

代码实现:

#include<iostream>
#include<queue>
#include<vector>
using namespace std;
const int N=1e5+5;
vector<int>v[N];
int level[N];
int idx;
void bfs(int root){
    queue<pair<int,int> >q;
    q.push({root,1});
    while(q.size()){
        int s=q.front().first;
        int dis=q.front().second;
        idx=max(idx,dis);
        q.pop();
        if(v[s].size()==0){
            level[dis]++;
            continue;
        }
        for(int i=0;i<v[s].size();i++){
            q.push({v[s][i],dis+1});
        }
    }
}
int main(){
    int n,m;
    cin>>n>>m;
    while(m--){
        string s;
        int k;
        cin>>s>>k;
        while(k--){
            string s1;
            cin>>s1;
            int x=stoi(s);
            int y=stoi(s1);
            v[x].push_back(y);
        }
    }
    bfs(1);
    for(int i=1;i<=idx;i++){
        if(i==1)cout<<level[i];
        else cout<<" "<<level[i];
    }
    return 0;
}
posted @ 2023-04-20 18:06  回忆、少年  阅读(12)  评论(0编辑  收藏  举报