PAT 1004 Counting Leaves (30 分) dfs

1004 Counting Leaves (30 分)

#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;
}