PAT-A 1004 Counting Leaves

题目大意

题目给出了一个树，要求由根节点往下，依次输出每一层的叶子节点个数。

解题思路

主要使用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;
}