【PAT 甲级】1004 Counting Leaves (30分)（dfs水）

stormjing7

于 2020-01-28 20:13:42 发布

阅读量203

点赞数

CC 4.0 BY-SA版权

分类专栏：解题报告 PAT 甲级 -----搜索-----

本文链接：https://blog.csdn.net/GodJing007/article/details/104101994

解题报告同时被 3 个专栏收录

178 篇文章

订阅专栏

-----搜索-----

22 篇文章

订阅专栏

PAT 甲级

11 篇文章

订阅专栏

本文介绍了一种算法，用于计算一棵节点数不超过100的树中，每一层的叶子节点数量。通过建立树结构并使用深度优先搜索（DFS），记录每个节点的层数及是否为叶子节点，最终统计并输出各层的叶子节点数目。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1004 Counting Leaves (30分)

题目

给一颗节点数不超过 100 的树，求每一层的叶子节点的个数。

分析

建树之后记录每个节点有无孩子，之后 dfs 一下记录每个节点在第几层。将信息归总统计一下即可得出每层叶子节点数。

#include <bits/stdc++.h>
using namespace std;
#define db(x) cout<<x<<endl

typedef long long ll;
const int INF = 0x3f3f3f3f;
const int N = 1e2 + 10;
const ll mod = 2147483648;

int n, m, cur, k, son;
int a[N][N];	// 建树
int has_son[N];	// 记录有无孩子
int level[N];	// 记录每个节点的层数
map<int, int> mp;	// 记录每层的答案

int dfs(int rt, int fa, int l) {
	level[rt] = l;	
	for (int i = 1; i <= n; i++) {
		if (i != fa && a[rt][i]) {
			dfs(i, rt, l + 1);
		}
	}
}

int main() {
	scanf("%d%d", &n, &m);
	while(m--) {
		scanf("%d%d", &cur, &k);
		while (k--) {
			scanf("%d", &son);
			a[cur][son] = 1;
			has_son[cur] = 1;
		}
	}
	dfs(1, 0, 1);
	for (int i = 1; i <= n; i++) {
		if (!has_son[i]) mp[level[i]]++;
		else mp[level[i]] += 0;
	}
	for (map<int, int>::iterator it = mp.begin(); it != mp.end(); it++) {
		printf("%d%c", it->second, " \n"[it->first==mp.rbegin()->first]);
	}
	return 0;	
}