PAT 甲级 1004

家谱统计

题目思路：

本题其实不用建立树的结构，可以把这个题目看成一个使用std::vector的DFS练习。在输入的时候，把每一层的编号看成以该层的根的编号，该层vector存储以该层为根，孩子的编号。那么，使用DFS就可以计算每一层的编号，需要用一个全局的数组ans进行存储。注意变量mm的使用，这是为了记录最大的结点编号，这种方式以后要留意下。

代码：

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 105;
int ans[MAXN], mm, N, M;
vector<int>G[MAXN];

void dfs(int u, int depth) {
    mm = max(mm, depth);
    if(G[u].size() == 0) {
        ++ans[depth]; // 如果是叶子结点，计数加1
        return;    
    }
    for(int i = 0; i < G[u].size(); ++i) {
        dfs(G[u][i], depth + 1);
    }
}

int main() {
    cin >> N >> M;
    for(int i = 0; i < M; ++i) {
        int id, k;
        cin >> id >> k;
        for(int j = 0; j < k; ++j) {
            int num;
            cin >> num;
            G[id].push_back(num);
        }
    }
    memset(ans, 0, sizeof(ans));
    mm = -1;
    dfs(1, 1);
    for(int i = 1; i < mm; ++i) {
        cout << ans[i] << " ";
    }
    cout << ans[mm] << endl;
    return 0;
}