PAT.A1004 Counting Leaves

返回目录

题意

给出一颗树，问有每层有多少叶子节点

样例（可复制）

2 1
01 1 02

//output
0 1

注意点

1. 本题可以使用DFS和BFS两种写法，下面采用的是DFS的写法

#include <bits/stdc++.h>
using namespace std;

vector<int> tree[100];
int N,M,child;
int numberoflevel[100];
int maxlevel=-1;
void DFS(int v,int level){
	maxlevel=max(level,maxlevel);
	if(tree[v].empty()){
		++numberoflevel[level];
	}
	for(int i:tree[v]){
		DFS(i,level+1);
	}
}
int main(){
	scanf("%d%d",&N,&M);
	while(M--){
		int id,k;
		scanf("%d%d",&id,&k);
		while(k--){
			scanf("%d",&child);
			tree[id].push_back(child);
		}
	}
    DFS(1,0);
	for(int i=0;i<=maxlevel;i++){
		if(i!=0)printf(" ");
		printf("%d",numberoflevel[i]);
	}
	return 0;
}