L2-051 满树的遍历（2024天梯赛）

一棵“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

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int n,maxans,flag,root,vis[N];
vector<int>v[N],ans;
void dfs(int root){
    ans.push_back(root);
    vis[root]=1;
    for(auto i:v[root]) if(!vis[i]) dfs(i);
}
int main(){
cin>>n;
for(int i=1;i<=n;i++){
    int a;
    cin>>a;
    if(!a) root=i;
    v[a].push_back(i);
}
for(int i=1;i<=n;i++){
    maxans=max(maxans,(int)v[i].size());
}
    cout<<maxans;
    for(int i=1;i<=n;i++){
        if(v[i].size()&&v[i].size()-maxans) flag=1;
    }
cout<<(flag? " no":" yes")<<endl;
dfs(root);
    cout<<ans[0];
    for(int i=1;i<ans.size();i++) cout<<" "<<ans[i];
    return 0;
}

 题目说是多叉树，其实就是简单的图遍历(dfs)