PAT甲题题解-1004. Counting Leaves (30)-统计每层叶子节点个数+dfs

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
using namespace std;
/*
统计每层的叶子节点个数
建树，然后dfs即可
*/
const int maxn=105;
int n,m;
int layer[maxn]; //统计每层的叶子节点个数
int maxlayer=0;
vector<int>node[maxn];

void dfs(int u,int level){
int num=node[u].size();
if(num==0){
layer[level]++;
if(level>maxlayer)
maxlayer=level;
return;
}
for(int i=0;i<node[u].size();i++){
dfs(node[u][i],level+1);
}
}
int main()
{
memset(layer,0,sizeof(layer));
scanf("%d %d",&n,&m);
int u,v,k;
for(int i=0;i<m;i++){
scanf("%d %d",&u,&k);
for(int i=0;i<k;i++){
scanf("%d",&v);
node[u].push_back(v);
}
}
dfs(1,0);
printf("%d",layer[0]);
for(int i=1;i<=maxlayer;i++)
printf(" %d",layer[i]);
return 0;
}