PAT(Advanced)1094 The Largest Generation 先序遍歷 C++實現

PAT(Advanced)甲級1094 The Largest Generation 先序遍歷 C++實現

題目鏈接

1094 The Largest Generation

題目大意

給定家族樹，求解人數最多的一代，輸出最多人數和那一代的代數，保證答案唯一

算法思路

根據題目要求和樹的先序遍歷定義，遍歷整棵樹，對應的層數累加，最後遍歷所有層的節點數得到最大值即可

void preOrderTraverse(int currentLevel, int order) {
    level[currentLevel]++;
    for (int i = 0; i < tree[order].size(); i++) {
        preOrderTraverse(currentLevel + 1, tree[order][i]);
    }
}

AC代碼

/*
author : eclipse
email  : [email protected]
time   : Sun Jun 28 11:35:09 2020
*/
#include <bits/stdc++.h>
using namespace std;

vector<vector<int> > tree;
vector<int> level;

void preOrderTraverse(int currentLevel, int order) {
    level[currentLevel]++;
    for (int i = 0; i < tree[order].size(); i++) {
        preOrderTraverse(currentLevel + 1, tree[order][i]);
    }
}

void print() {
    int levelOrder = 0;
    for (int i = 1; i < level.size(); i++) {
        if (level[levelOrder] < level[i]) {
            levelOrder = i;
        }
    }
    printf("%d %d", level[levelOrder], levelOrder);
}

int main(int argc, char const *argv[]) {
    int N, M;
    scanf("%d%d", &N, &M);
    tree.resize(N + 1);
    for (int i = 0; i < M; i++) {
        int ID, K;
        scanf("%d%d", &ID, &K);
        for (int j = 0; j < K; j++) {
            int child;
            scanf("%d", &child);
            tree[ID].push_back(child);
        }
    }
    level.resize(N + 1);
    preOrderTraverse(1, 1);
    print();
    return 0;
}

樣例輸入

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

鳴謝

PAT

最後

• 由於博主水平有限，不免有疏漏之處，歡迎讀者隨時批評指正，以免造成不必要的誤解！