PAT 甲级 1004 Counting Leaves (30 分)

两种做法:
1、使用队列进行BFS,层序遍历,每层开始时设置一个size记录当前层的结点数,每遍历完一层就输出一层的结果,最后一层结束后队列为空作为结束条件和格式输出条件
2、设置一个下标表示深度的数组,进行dfs,每遇到一个叶子结点就在对应深度加1,计算最大深度是为了最后的输出

Code1:

#include<iostream> 
#include<vector>
#include<queue>

using namespace std;

vector<int> node[101];
queue<int> q;

int main() {
	int n, m;
	cin >> n >> m;
	for (int i = 0; i < m; i++)
	{
		int p, num, c;
		cin >> p >> num;
		for (int j = 0; j < num; j++)
		{
			cin >> c;
			node[p].push_back(c);
		}
	}
	q.push(1);
	while (!q.empty())
	{
		int size = q.size();
		int cnt = 0;
		for (int i = 0; i < size; i++)
		{
			int parent = q.front();
			q.pop();
			if (node[parent].empty()) cnt++;
			else {
				for (int j = 0; j < node[parent].size(); j++)
				{
					q.push(node[parent][j]);
				}
			}
		}
		if (q.empty()) {
			cout << cnt;
		}
		else {
			cout << cnt << " ";
		}
		
	}
	return 0;
}

Code2:

#include<iostream>
#include<vector>

using namespace std;

vector<int> v[100];
int num[100];//存储每层的叶子节点个数
int max_depth = -1;

void dfs(int node, int depth)
{
	if (v[node].size() == 0)
	{
		num[depth]++;
		max_depth = max_depth > depth ? max_depth : depth;
		return;
	}
	int cnt = v[node].size();
	for (int i = 0; i < cnt; i++)
	{
		dfs(v[node][i], depth + 1);
	}
}

int main()
{
	int N;//总结点数
	int M;//非叶子节点数
	int k;//非叶子结点的孩子个数
	int node;//非叶子结点编号
	int c;//孩子节点编号
	scanf("%d %d", &N, &M);
	for (int i = 0; i < M; i++)
	{
		scanf("%d %d", &node, &k);
		for (int j = 0; j < k; j++)
		{
			scanf("%d", &c);
			v[node].push_back(c);
		}
	}
	dfs(1,0);//root节点的编号为1,所在深度为0
	printf("%d", num[0]);
	for (int i = 1; i <= max_depth; i++)
	{
		printf(" %d", num[i]);
	}
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值