1004. Counting Leaves (30)--dfs

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <cmath>

#include <string>
#include <cstring>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <stack>

using namespace std;

int N,M;
struct node
{
    vector<int> child;
}cube[200];

int book[200]={0};
int maxx=-1;
void dfs(int id,int depth)
{
    if(cube[id].child.size()==0)//叶子
    {
        book[depth]++;
        if(depth>maxx)
        {
            maxx=depth;
        }
        return;
    }

    for(int i=0;i<cube[id].child.size();i++)
    {
        dfs(cube[id].child[i],depth+1);
    }

    return;
}

int main()
{
//    freopen("in.txt","r",stdin);

    scanf("%d %d",&N,&M);
    for(int i=0;i<M;i++)
    {
        int id;
        int n;
        scanf("%d %d",&id,&n);
        for(int j=0;j<n;j++)
        {
            int t;
            scanf("%d",&t);
            cube[id].child.push_back(t);
        }
    }

    dfs(1,0);

    for(int i=0;i<=maxx;i++)
    {
        printf("%d",book[i]);
        if(i!=maxx)
            printf(" ");
    }



    return 0;
}