PTA数据结构：满树的遍历

满树的遍历

题目难度： ☀️

题目描述：

一棵“k 阶满树”是指树中所有非叶结点的度都是 k 的树。给定一棵树，你需要判断其是否为 k 阶满树，并输出其前序遍历序列。

注：树中结点的度是其拥有的子树的个数，而树的度是树内各结点的度的最大值。

输入格式：

输入首先在第一行给出一个正整数 n（≤105），是树中结点的个数。于是设所有结点从 1 到 n 编号。
随后 n 行，第 i 行（1≤i≤n）给出第 i 个结点的父结点编号。根结点没有父结点，则对应的父结点编号为 0。题目保证给出的是一棵合法多叉树，只有唯一根结点。

输出格式：

首先在一行中输出该树的度。如果输入的树是 k 阶满树，则加 1 个空格后输出 yes，否则输出 no。最后在第二行输出该树的前序遍历序列，数字间以 1 个空格分隔，行首尾不得有多余空格。
注意：兄弟结点按编号升序访问。

输入样例 1：

7
6
5
5
6
6
0
5

输出样例 1：

3 yes
6 1 4 5 2 3 7

输入样例 2：

7
6
5
5
6
6
0
4

输出样例 2：

3 no
6 1 4 7 5 2 3

解题思路：

默认读者已经自行了解何为树的先序。简单的先序遍历，dfs

以下是完整AC代码：

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
const int N = 1e5 + 10;
vector<vector<int> > tree(N); int root;
vector<int> ans; vector<int> vir(N,0);
// tree数组的下标作为节点，tree[i]中的值表示他的子节点
// root是整棵树的根节点，即root的父节点是0
// ans是先序遍历得到的答案数组
// vir表示每个节点在dfs时是否已经访问，即0为未访问，注意初始化。
void dfs(int root){
    ans.push_back(root); 
    vir[root] = 1;
    // 根节点先输出，然后标记根节点
    for(auto i : tree[root]) {
        if(!vir[i]) dfs(i); 
    }
    // 若此节点未被访问，dfs，查看根节点的子节点是否有自己的子节点
}

int main(){
    ios::sync_with_stdio(false); cin.tie(0); cout.tie(0); // 输入输出的维护
	int n; cin >> n;
    for(int i = 1; i <= n; i++){
        int father; cin >> father;
        if(!father) root = i;
        // 找到根节点
        tree[father].push_back(i);
        // 在根节点中加入他的子节点
    }
    int maxn = 0;
    // 求整棵树的度，即某个根节点所拥有的字节点数量最大
    for(int i = 1; i <= n; i++){
        maxn = max(maxn, (int)tree[i].size());
    }
    cout << maxn << " ";
    // 输出树的度
    bool flag = true;
    // flag表示树是否是满阶树
    for(int i = 1; i <= n; i++){
		if((int)tree[i].size() && (int)tree[i].size() < maxn) flag = false;
        // 每个根节点拥有的子节点的数量，若不等于树的度，就不是满阶树
    }
    cout << (flag ? "yes" : "no") << "\n";
    dfs(root);
    // 先序遍历
    cout << ans[0];
    for(int i = 1; i < (int)ans.size(); i++){
        cout << " " << ans[i];
    }
    // 输出答案，注意输出格式，没有多余空格和空行
    return 0;
}