PAT甲级1004 Counting Leaves(DFS / BFS)

传送门：1004 Counting Leaves

题意：对于一棵树，给定树的结点个数n（＜100）与内部节点（非叶子结点）个数m（＜n），以及m个内部结点的孩子结点，以内部节点的id，内部节点id的孩子数量k，内部节点id的k个孩子的输入形式给出。特别的，01总是树的根节点。要求的是，每一层的叶子结点的数量。

思路：先构造出给定的树，然后从根结点开始遍历，记录叶子结点的深度，遍历完成后，每一层的叶子数便计算完成。

关于遍历，博主采用的是DFS，如果顺着题意来，也可以使用层次遍历（BFS）。

细节：①题目要求当n为0时直接跳过，不进行处理；②维护最大深度max_depth。

#include <bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
const int maxn = 1e2;
int n, m;
vector<int> child[maxn]; //存储每个结点的孩子结点
int ans[maxn], max_depth = -INF;

bool read()
{
    cin >> n >> m;
    if(!n) return false;

    for(int i = 0; i < m; ++i)
    {
        int id, k;
        cin >> id >> k;
        for(int j = 0; j < k; ++j)
        {
            int id_child;
            cin >> id_child;
            child[id].push_back(id_child);
        }
    }
    return true;
}

void dfs(int rt, int depth)
{
    int n_child = child[rt].size();
    if(!n_child)
    {
        ++ans[depth];
        max_depth = max(max_depth, depth); //维护最大深度
        return;
    }
    for(int i = 0; i < n_child; ++i)
        dfs(child[rt][i], depth+1); //当前结点的孩子结点深度+1
}

void solve()
{
    dfs(1, 0); //根节点深度为0
    cout << ans[0];
    for(int i = 1; i <= max_depth; ++i)
        cout << ' ' << ans[i];
}

int main()
{
    if(read()) solve();
    return 0;
}