PAT(Advanced Level) 1004 Counting Leaves

给出结点数 n，非叶子结点 m，以下 m 行给出非叶子结点的数值、孩子数量 k 以及 k 个孩子结点的数值，问每一层各有多少个叶子结点 ？

使用 DFS，用 vector 数组保存每一个非叶子结点以及其所有的孩子结点，从根结点开始遍历，直至遇到叶子结点（ 判断条件：id[i].size()==0 ），就将当前层数 depth 的 leafnum[depth]++ ，同时更新树的深度 maxdepth

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
vector<int> id[100];
int leafnum[100],maxdepth=-1;

void dfs(int root,int depth){
    if(id[root].size()==0){
        leafnum[depth]++;
        maxdepth = max(maxdepth,depth);
        return;
    }
    else{
        for(int i=0;i<id[root].size();i++)
            dfs(id[root][i],depth+1);
    }
}
int main(){
    int n,m;
    scanf("%d %d",&n,&m);
    int fm,k,child;
    for(int i=0;i<m;i++){
        scanf("%d %d",&fm,&k);
        for(int j=0;j<k;j++){
            scanf("%d",&child);
            id[fm].push_back(child);
        }
    }
    dfs(1,0);
    for(int i=0;i<=maxdepth;i++)
        if(i==0) printf("%d",leafnum[i]);
        else printf(" %d",leafnum[i]);
}