PAT A1094 The Largest Generation (25 分)

题意

给出树形的家谱，求人最多的一代。题目给出每个有孩子的家庭成员的孩子。

注意点

• 层数和根节点都是从1开始计算
• 树的结点结构体中需要一个数据域存放所在层数
• 用一个数组存放每层的人数

题解

解题思路：
①定义结点，其中有所在层数信息
②读入每个结点的孩子信息，创建静态二叉树
③BFS函数，将每层的人数累加到numlayer数组中
④遍历numlayer数组，找出人数最多的一层，输出人数及层数

#include <stdio.h>
#include <queue>
#include <vector>
using namespace std;
const int maxn = 110;
//定义结点
struct node {
    int layer;
    vector<int> child;
} Node[maxn];
//变量
int n, m;
//BFS
int maxlayer = 1, numLayer[maxn] = {0};
void BFS(int root) {
    queue<int> q;
    q.push(root);
    while (!q.empty()) {
        int now = q.front();
        q.pop();
        int num = 0;
        for (int i = 0; i < Node[now].child.size(); i++) {
            int child;
            child = Node[now].child[i];
            Node[child].layer = Node[now].layer + 1;
            q.push(child);
            numLayer[Node[child].layer]++;
        }
        if (Node[now].layer + 1 > maxlayer) {
            maxlayer = Node[now].layer + 1;
        }
    }
}
int main() {
    scanf("%d%d", &n, &m);
    for (int i = 0; i < m; i++) {
        int member, childNum;
        scanf("%d%d", &member, &childNum);
        for (int j = 0; j < childNum; j++) {
            int nowchild;
            scanf("%d", &nowchild);
            Node[member].child.push_back(nowchild);
        }
    }
    int root = 1;
    Node[root].layer = 1;
    numLayer[1] = 1;
    BFS(1);
    int max = 0, layer = 0;
    for (int i = 1; i <= maxlayer; i++) {
        if (numLayer[i] > max) {
            max = numLayer[i];
            layer = i;
        }
    }
    printf("%d %d\n", max, layer);
    return 0;
}