【题解】【AcWing】1476. 数叶子结点

186 阅读2分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

1476. 数叶子结点

原题传送:AcWing 1476. 数叶子结点

家庭关系可以用家谱树来表示,给定一个家谱树,你的任务是找出其中没有孩子的成员。

输入格式

第一行包含一个整数 NN 表示树中结点总数以及一个整数 MM 表示非叶子结点数。

接下来 MM 行,每行的格式为:

ID K ID[1] ID[2] ... ID[K]

IDID 是一个两位数字,表示一个非叶子结点编号, KK 是一个整数,表示它的子结点数,接下来的 KKID[i]ID[i] 也是两位数字,表示一个子结点的编号。

为了简单起见,我们将根结点固定设为 0101

所有结点的编号即为 01,02,03,,31,32,33,,N01,02,03,…,31,32,33,…,N

输出格式

输出从根结点开始,自上到下,树的每一层级分别包含多少个叶子节点。

输出占一行,整数之间用空格隔开。

数据范围

0<N<1000<N<100

输入样例:

2 1
01 1 02

输出样例:

0 1

样例解释

该样例表示一棵只有 22 个结点的树,其中 0101 结点是根,而 0202 结点是其唯一的子节点。

因此,在根这一层级上,存在 00 个叶结点;在下一个级别上,有 11 个叶结点。

所以,我们应该在一行中输出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;
}