1004 Counting Leaves

1004 Counting Leaves

题目大意

根据特定的输入，要求每一行的叶结点数量

算法思想

• 一眼看去，大概率会想到广度优先，但是仔细想想，需要设置队列和判断为第几层，还是深度优先更好些
• 二维数组存储每个结点的孩子，这个貌似挺特别
• 深度优先时，需要得到该树有多高
• 没什么特别注意的点，能写出来就是满分

代码

#include<iostream>
#include<vector>
using namespace std;
int all, nleaf;
vector<int>deep(100);//每一层的叶结点数量
int maxh = 0;
void dfs(vector<vector<int>> v,int now,int h)//深度优先遍历
{
	if (h > maxh)//找出该树的高度
	{
		maxh = h;
	}
	if (v[now].size() == 0)//判断如果为叶结点，该行加1
	{
		deep[h]++;
	}
	else {
		for (int i = 0; i < v[now].size(); i++)//如果为非叶结点，深度遍历其孩子
		{
			dfs(v, v[now][i], h + 1);
		}
	}
	return;
}
int main()
{
	int all, nleaf;
	cin >> all >> nleaf;
	vector<vector<int>>tree(all);//用个二维数组来存储
	for (int i = 0; i < 100; i++)//初始化为0，方便自增
	{
		deep[i] = 0;
	}
	for (int i = 0; i < nleaf; i++)//循环输入
	{
		int node, k;
		cin >> node >> k;
		for (int j = 0; j < k; j++)
		{
			int p;
			cin >> p;
			tree[node-1].push_back(p-1);//因为从0开始，要将输入的数减1
		}
	}
	dfs(tree, 0, 0);//开始深度遍历
	int i;
	for (i = 0; i < maxh; i++)//需要格式输出
	{
		cout << deep[i] << " ";
	}
	cout << deep[i];

}