1004. Counting Leaves (30)

用vector存储每个非叶子节点的孩子，再一次dfs找出叶子节点以及最大深度。

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <vector>
#include <map>
#include <stack>
#include <queue>
#include <algorithm>
#include <set>
using namespace std;
struct node{
    vector<int> child;
};
node tree[105];
int deepnum[100]={0},maxdeep=0;
void DFS(int id,int deep){
    deep++;
    if(deep>maxdeep)
        maxdeep=deep;
    if(tree[id].child.empty())
        deepnum[deep]++;
    else{
        for(int i=0;i<tree[id].child.size();i++)
            DFS(tree[id].child[i],deep);
    }
}
int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=0;i<m;i++){
        int id,c,k;
        //node tr;
        scanf("%d%d",&id,&c);
        while(c--){
            scanf("%d",&k);
            tree[id].child.push_back(k);
        }
    }
    int id=1,deep=0;
    DFS(id,deep);
    printf("%d",deepnum[1]);
    for(int i=2;i<=maxdeep;i++)
        printf(" %d",deepnum[i]);
    return 0;
}