#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
vector<int> v[100];
int book[100]={0},maxdepth=-1; //book[depth]标记第depth层拥有的叶⼦结点数
void dfs(int index,int depth){
if(v[index].size()==0){ //到达底部
book[depth]++;
maxdepth=max(depth,maxdepth);
return;
}
for(int i=0;i<v[index].size();i++){
dfs(v[index][i],depth+1);
}
}
int main(){
int n,m,node,k; //其中0<N<100表示树中的节点数,M(<N)表示非叶节点数
cin>>n>>m;
for(int i=0;i<m;i++){ //输入每个结点的孩子
scanf("%d %d",&node,&k);
v[node].resize(k);
for(int j=0;j<k;j++){
scanf("%d",&v[node][j]);
}
}
dfs(1,0);
printf("%d",book[0]);
for(int i=1;i<=maxdepth;i++){
printf(" %d",book[i]);
}
system("pause");
return 0;
}
PAT 1004 Counting Leaves (30 分) dfs
于 2022-02-04 17:13:11 首次发布