PAT1004. Counting Leaves bfs

这个题目的意思是给出这棵树的节点数和非叶节点，统计各层的叶节点数。

这道题目还是蛮简单的，虽然是给出一棵树，但是并不需要完整的建树，只需要知道每个节点有没有孩子和该节点的层数即可。

代码如下

#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<cstring>
using namespace std;
vector<int> tree[120];
int levelnum[120]; //节点层数
int ans[120];//结果 每层的非叶节点数量
// 字符串转整数
int chattoInt(char a[2]){
int b =(a[0]-'0')*10;
int c = a[1]-'0';
return b+c;
}

void solver(){
queue<int> q;
q.push(1);
levelnum[1] = 1;
int level=0;
memset(ans,0,sizeof(ans));
while(!q.empty())
{
int t = q.front();
q.pop();
if(levelnum[t]>level)
level = levelnum[t];
int cou = tree[t].size();
if(cou==0)
{
ans[level]++;
}
else{
for(int i=0;i<tree[t].size();i++)
{
q.push(tree[t][i]);
levelnum[tree[t][i]] = levelnum[t]+1;
}
}

}
}
int main(){
int m,n;
cin>>n>>m;
for(int i=0;i<m;i++)
{
char id[2];
int k;
cin>>id[0]>>id[1]>>k;
int id1 = chattoInt(id);
for(int j=0;j<k;j++)
{
char id2[2];
cin>>id2[0]>>id2[1];
int id3 = chattoInt(id2);
tree[id1].push_back(id3);

}
}
solver();
int sum=0;
cout<<ans[1];
for(int i=2;i<=n;i++) //最后好几点没过就是因为输出结构没控制好
{
if(sum!=n-m){
cout<<" "<<ans[i];
sum+=ans[i];
}

}
return 0;
}