PAT甲级 1094 最大的一代

原题链接

家庭关系可以用家谱树来表示,同一层上的所有结点都属于同一代人。

请你找出人数最多的一代。

输入格式
第一行包含一个整数 N 表示树中结点总数以及一个整数 M 表示非叶子结点数。

接下来 M 行,每行的格式为:

ID K ID[1] ID[2] ... ID[K]
ID 是一个两位数字,表示一个非叶子结点编号,K 是一个整数,表示它的子结点数,接下来的 K 个 ID[i] 也是两位数字,表示一个子结点的编号。

为了简单起见,我们将根结点固定设为 01。

所有结点的编号即为 01,02,03,…,31,32,33,…,N。

输出格式
输出结点数量最多的一层的结点数量以及层级。

保证答案唯一。

根结点定义为第一层。

数据范围
0<N<100
输入样例:
23 13
21 1 23
01 4 03 02 04 05
03 3 06 07 08
06 2 12 13
13 1 21
08 2 15 16
02 2 09 10
11 2 19 20
17 1 22
05 1 11
07 1 14
09 1 17
10 1 18
输出样例:
9 4

我的解法:

#include <bits/stdc++.h>
using namespace std;
const int N = 110;
bool g[N][N];
int n, m;
vector <int> lev[N];
int main(){
    cin >> n >> m;
    for(int i = 0; i < m; i ++ ){
        int k, id;
        cin >> id >> k;
        while(k -- ){
            int son;
            cin >> son;
            g[id][son] = true;
        }
    } 
    int l = 1;
    lev[1].push_back(1);
    while(lev[l].size()){
        for(auto p : lev[l]){
            for(int i = 1; i <= n; i ++ ){
                if(g[p][i]) lev[l + 1].push_back(i);
            }
        }

        l ++;
    }
    int k = 1;
    for(int i = 1; i < l; i ++ ){
        if(lev[i].size() > lev[k].size()) k = i;
    }
    cout << lev[k].size() << ' ' << k << endl;
    return 0;
}

收获:

灵活运用bfs

猜你喜欢

转载自blog.csdn.net/weixin_45660485/article/details/124918182