题意:对于一棵树,给定树的结点个数n(<100)与内部节点(非叶子结点)个数m(<n),以及m个内部结点的孩子结点,以内部节点的id,内部节点id的孩子数量k,内部节点id的k个孩子的输入形式给出。特别的,01总是树的根节点。要求的是,每一层的叶子结点的数量。
思路:先构造出给定的树,然后从根结点开始遍历,记录叶子结点的深度,遍历完成后,每一层的叶子数便计算完成。
关于遍历,博主采用的是DFS,如果顺着题意来,也可以使用层次遍历(BFS)。
细节:①题目要求当n为0时直接跳过,不进行处理;②维护最大深度max_depth。
#include <bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
const int maxn = 1e2;
int n, m;
vector<int> child[maxn]; //存储每个结点的孩子结点
int ans[maxn], max_depth = -INF;
bool read()
{
cin >> n >> m;
if(!n) return false;
for(int i = 0; i < m; ++i)
{
int id, k;
cin >> id >> k;
for(int j = 0; j < k; ++j)
{
int id_child;
cin >> id_child;
child[id].push_back(id_child);
}
}
return true;
}
void dfs(int rt, int depth)
{
int n_child = child[rt].size();
if(!n_child)
{
++ans[depth];
max_depth = max(max_depth, depth); //维护最大深度
return;
}
for(int i = 0; i < n_child; ++i)
dfs(child[rt][i], depth+1); //当前结点的孩子结点深度+1
}
void solve()
{
dfs(1, 0); //根节点深度为0
cout << ans[0];
for(int i = 1; i <= max_depth; ++i)
cout << ' ' << ans[i];
}
int main()
{
if(read()) solve();
return 0;
}