题106.pat甲级练习-1004 Counting Leaves (30 分)
一、题目
二、题解
本题要你去数树的每一层的叶节点个数,我们可以通过改造bfs来解决这个问题,核心操作就是碰到每一层的叶节点时对用于记录每一层叶节点树的数组的元素加1
#include <bits/stdc++.h>
using namespace std;
int N,M;
int G[101][101];
int visited[101];
int level[101];//记录每个节点所在的层级
int res[101];//记录每个层级的叶子节点数目
int bfs(int root)//从根节点开始bfs
{
queue<int> q;
int nowlevel=1;
q.push(root);
level[root]=1;//根节点位于1层
visited[root]=1;
while(!q.empty())
{
//cout<<"1"<<endl;
int temp=q.front();
q.pop();
int flag=1;
nowlevel=level[temp];//当前层级设为队头节点的层级
for(int v=1;v<=N;v++)
{
if(G[temp][v]==1&&visited[v]==0)
{
flag=0;
q.push(v);
level[v]=nowlevel+1;//这是队头节点的儿子,他们是当前层级的下一层级的
visited[v]=1;
}
}
if(flag)
{
res[nowlevel]++;//只要没有儿子节点,那么当前层级叶子节点数加1
}
}
return nowlevel;
}
int main()
{
cin>>N>>M;
for(int i=0;i<M;i++)
{
int ID,K;
scanf("%d%d",&ID,&K);
for(int j=0;j<K;j++)
{
int Child_ID;
scanf("%d",&Child_ID);
G[ID][Child_ID]=1;//注意这是树,所以只能单向连通
}
}
int levels=bfs(1);
for(int i=1;i<=levels;i++)
{
printf("%d",res[i]);
if(i<levels)
{
putchar(' ');
}
}
}
相比于整体思路的构建,操作里头如何知道当前节点层级这个问题是更让我头大的,因为之前有写过类似的操作,但是每次都费劲,毕竟脑子实在不好。但这次我自己终于想到了一个不会让我费劲的知道当前层级的路子了,我可以用nowlevel来记录当前层级,然后给每一个节点整一个所在层级,用数组存着,开始的时候根节点肯定是第一层,然后之后队列pop出来便直接进入下一层(因为根节点就它一个),这时和它相连的节点的level就是第二层(当前层级加1),之后也是如此,只是一层的节点数可能变多了而已。这样一直往下一层走,当前层级也会不断改变·,而每个节点所在层级也会得出,二者相互依靠。