原题传送门

1. 问题描述

2. Solution

1、思路分析
题意: 给出一棵树，问每一层各有多少叶子结点。
分析: 可以用dfs也可以用bfs。以下用dfs实现，用二维数组保存每一个孩子结点的结点以及他们的孩子结点，从根结点开始遍历，直到遇到叶子结点，就将当前层数depth的leaf_count[depth]++；标记第depth层拥有的叶子结点数，最后输出。

2、代码实现

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

vector<int> v[100];
int leaf_count[100], max_depth = -1;

void dfs(int index, int depth) {
    if (v[index].empty()) {
        leaf_count[depth]++;
        max_depth = max(max_depth, depth);
        return;
    }
    for (int i = 0; i < v[index].size(); i++)
        dfs(v[index][i], depth + 1);
}

int main() {
#ifdef ONLINE_JUDGE
#else
    freopen("./input/1004.txt", "r", stdin);
#endif
    int n, m, k, node, c;
    scanf("%d %d", &n, &m);
    for (int i = 0; i < m; i++) {
        scanf("%d %d", &node, &k);
        for (int j = 0; j < k; j++) {
            scanf("%d", &c);
            v[node].push_back(c);
        }
    }
    dfs(1, 0);
    printf("%d", leaf_count[0]);
    for (int i = 1; i <= max_depth; i++)
        printf(" %d", leaf_count[i]);
    return 0;
}

标签：结点,PAT,Level,int,dfs,++,depth,leaf,1004
来源： https://www.cnblogs.com/junstat/p/16188605.html