1004 Counting Leaves

最新推荐文章于 2024-11-12 12:29:51 发布

天金小麻花

最新推荐文章于 2024-11-12 12:29:51 发布

阅读量234

点赞数

本文链接：https://blog.csdn.net/ATCG7/article/details/125525009

版权

深度优先搜索 DFS 树结构叶子节点遍历算法

关键词由CSDN通过智能技术生成

在这里插入图片描述
代码：

#include<iostream>
#include<vector>
using namespace std;

vector<int> v[1000];
int leafnum[1000];//每层叶子节点个数 
int maxdep=-1;

void dfs(int node,int dep){
	//如果该节点的子节点个数为0，说明是叶节点 
	if(v[node].size()==0){
		leafnum[dep]++;
		if(dep>maxdep){
			maxdep=dep;
		} 
		return;
	}
	for(int i=0;i<v[node].size();i++){
		dfs(v[node][i],dep+1);
	}
}
int main(){
	int n,m,id,k;
	cin>>n>>m;
	while(m--){
		cin>>id>>k;
		while(k--){
			int temp;
			cin>>temp;
			v[id].push_back(temp);
		}
	}
	dfs(1,0);
	cout<<leafnum[0];
	for(int i=1;i<=maxdep;i++){
		cout<<" "<<leafnum[i];
	}
	;
	
}