【PAT甲级】1004 Counting Leaves (30 分)（BFS）

题意：给出一棵树的点数N，输入M行，每行输入父亲节点An，儿子个数n，和a1，a2,...,an（儿子结点编号），从根节点层级向下依次输出当前层级叶子结点个数，用空格隔开。（0<N<100)

代码：

#include<bits/stdc++.h>
using namespace std;
vector<int>v[107];
queue<int>q;
int tmp=0;
int num=0;
int cnt=0;
int tim=0;
int tt=0;
void BFS(){
int x=q.front();//队头元素调用
if(v[x].size()){
for(auto it:v[x]){//遍历x的所有儿子结点
if(v[it].size()==0)
num++;//叶子结点++
else{
q.push(it);//非叶子结点进队列
cnt++;//非叶子结点计数器++
}
}
}
q.pop();//队头元素弹出
tim++;//处理次数++
}
int main(){
int n,m;
cin>>n>>m;
for(int i=1;i<=m;++i){
string fa;
cin>>fa;
int baba=(fa[0]-'0')*10+fa[1]-'0';
int x;
cin>>x;
string son;
for(int j=1;j<=x;++j){
cin>>son;
int erzi=(son[0]-'0')*10+son[1]-'0';
v[baba].push_back(erzi);
}
}
if(v[1].size()==0){
cout<<"1";
return 0;
}
else
cout<<"0";
for(auto it:v[1]){
if(v[it].size()==0)
num++;//叶子结点个数
else{
q.push(it);//非叶子结点进队列
tt++;//非叶子结点个数
}
}
cout<<" "<<num;
num=0;
while(!q.empty()){
BFS();
if(tim==tt){//处理次数==非叶子结点个数，相当于这一层处理完了，开始统计这一层下一层有多少叶子结点
cout<<" "<<num;
tim=0;//处理次数初始化
num=0;//叶子结点个数初始化
tt=cnt;//非叶子结点个数更新为计数器
cnt=0;//非叶子节点计数器初始化
}
}
return 0;
}

转载于:https://www.cnblogs.com/ldudxy/p/11207198.html