题目大意
题目给出了一个树,要求由根节点往下,依次输出每一层的叶子节点个数。
解题思路
主要使用bfs,遍历树的每一层的所有节点,计算所有叶子节点的数目即可。每个节点的子节点可以用一个数组来保存。
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
int main()
{
vector<vector<int>> v(100);
int N,M;
cin>>N>>M;
for(int i=0;i<M;++i)
{
int id,K;
cin>>id>>K;
for(int j=0;j<K;++j)
{
int temp;
cin>>temp;
v[id].push_back(temp);
}
}
//bfs
queue<int> q;
q.push(1);
int size=q.size(),cnt=0;
bool first=true;
while(!q.empty())
{
int temp=q.front();
q.pop();
if(v[temp].empty())
++cnt;
for(int i=0;i<v[temp].size();++i)
q.push(v[temp][i]);
if(--size==0)
{
if(first)
{
cout<<cnt;
first=false;
}
else
cout<<" "<<cnt;
size=q.size();
cnt=0;
}
}
return 0;
}