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;
}