PAT甲级 1004 树

思路：直接遍历整棵树判定每个结点是否有孩子，没有则把当前高度的叶子节点数加一。

---

AC代码

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <vector>
using namespace std;
const int maxn = 100+5;
vector<int> G[maxn];
int level[maxn];
int height;

void dfs(int u, int lev) {
    height = max(height, lev);
    if(G[u].size() == 0) {
        level[lev]++;
        return;
    } 
    for(int i = 0, n = G[u].size(); i < n; i++) {
        int v = G[u][i];
        dfs(v, lev+1);
    }
}

int main() {
    int n, m, k;
    int u, v;
    scanf("%d %d", &n, &m);
    for(int i = 0; i < m; i++) {
        scanf("%d %d", &u, &k);
        //printf("%d %d\n", u, k);
        for(int j = 0; j < k; j++) {
            scanf("%d", &v);
            G[u].push_back(v);
        }
    }
    height = -1;
    memset(level, 0, sizeof(level));
    dfs(1, 0);
    for(int i = 0; i <= height; i++) {
        printf("%d%c", level[i], i == height ? '\n' : ' ');
    }
    return 0;
}

如有不当之处欢迎指出！