题意
给出树形的家谱,求人最多的一代。题目给出每个有孩子的家庭成员的孩子。
注意点
- 层数和根节点都是从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;
}