本文已参与「新人创作礼」活动,一起开启掘金创作之路。
1476. 数叶子结点
原题传送:AcWing 1476. 数叶子结点
家庭关系可以用家谱树来表示,给定一个家谱树,你的任务是找出其中没有孩子的成员。
输入格式
第一行包含一个整数 表示树中结点总数以及一个整数 表示非叶子结点数。
接下来 行,每行的格式为:
ID K ID[1] ID[2] ... ID[K]
是一个两位数字,表示一个非叶子结点编号, 是一个整数,表示它的子结点数,接下来的 个 也是两位数字,表示一个子结点的编号。
为了简单起见,我们将根结点固定设为 。
所有结点的编号即为 。
输出格式
输出从根结点开始,自上到下,树的每一层级分别包含多少个叶子节点。
输出占一行,整数之间用空格隔开。
数据范围
输入样例:
2 1
01 1 02
输出样例:
0 1
样例解释
该样例表示一棵只有 个结点的树,其中 结点是根,而 结点是其唯一的子节点。
因此,在根这一层级上,存在 个叶结点;在下一个级别上,有 个叶结点。
所以,我们应该在一行中输出01。
思路:
邻接链表存储树,dfs记录每层的叶结点个数。
题解:
#include<bits/stdc++.h>
using namespace std;
const int N = 110;
int n, m;
int h[N], e[N], ne[N], idx;
int cnt[N], max_depth;
void add(int a, int b)
{
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
void dfs(int u, int depth)
{
if(h[u] == -1)
{
cnt[depth]++;
max_depth = max(max_depth, depth);
return;
}
for(int i = h[u]; ~i; i = ne[i])
dfs(e[i], depth + 1);
}
int main()
{
cin >> n >> m;
memset(h, -1, sizeof h);
for(int i = 0; i < m; i++)
{
int id, k;
cin >> id >> k;
while(k--)
{
int son;
cin >> son;
add(id, son);
}
}
dfs(1, 0);
cout << cnt[0];
for(int i = 1; i <= max_depth; i++)
cout << " " << cnt[i];
cout << endl;
return 0;
}