pat甲级 1004 Counting Leaves（输出每层叶子结点）（dfs）

最新推荐文章于 2022-07-26 16:28:25 发布

原创最新推荐文章于 2022-07-26 16:28:25 发布 · 248 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pat甲级

pat甲级专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一个使用C++编程语言和STL中的vector容器实现深度优先搜索(DFS)算法的例子。该程序首先读取节点和边的数量，然后输入节点之间的连接关系，最后通过DFS遍历整个图并统计各层的节点数。本文展示了如何利用vector来存储图的邻接列表，并通过递归方式实现DFS算法。

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
vector<int> v[100];
int maxdepth = -1,book[105];
void dfs(int index, int depth) {
	if (v[index].size() == 0) {
		book[depth]++;
		maxdepth = max(maxdepth, depth);
		return;
	}
	for (int i = 0; i < v[index].size(); i++)
		dfs(v[index][i], depth + 1);
}
int main() {
	int n, m, kid, node, c;
	cin >> n >> m;
	for (int i = 0; i < m; i++) {
		cin >> node >> kid;
		for (int j = 0; j < kid; j++) {
			cin >> c;
			v[node].push_back(c);
		}
	}
	dfs(1, 0);
	cout << book[0];
	for (int i = 1; i <= maxdepth; i++) {
		cout << " " << book[i];
	}
	cout << endl;
	return 0;
}

vector好方便啊！！！！！！！！！