PAT.A1004葉のカウント

目次に戻る

タイトル

ツリーが与えられたら、各レイヤーにあるリーフノードの数を尋ねます

サンプル（コピー可能）

2 1
01 1 02

//output
0 1

注意点

1. この質問では、DFSとBFSの2つの書き込み方法を使用できます。以下では、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;
}