PAT (Advanced Level) 1004. Counting Leaves (30)

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdio>
#include<vector>
using namespace std;

const int maxn=100+10;
vector<int>g[maxn];
int n,m;
int ans[maxn];
int root;
int Deep;

void dfs(int x,int deep)
{
Deep=max(Deep,deep);
if(g[x].size()==0)
{
ans[deep]++;
return ;
}
for(int i=0;i<g[x].size();i++)
dfs(g[x][i],deep+1);
}

int main()
{
scanf("%d%d",&n,&m);
for(int i=0;i<=n;i++) g[i].clear();
for(int i=1;i<=m;i++)
{
int id; scanf("%d",&id);
int k; scanf("%d",&k);
while(k--)
{
int id2; scanf("%d",&id2);
g[id].push_back(id2);
}
}
memset(ans,0,sizeof ans);
root=1; Deep=1;
dfs(root,1);
for(int i=1;i<=Deep;i++)
{
printf("%d",ans[i]);
if(i<Deep) printf(" ");
else printf("\n");
}
return 0;
}