L2-3 满树的遍历

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

 solution:
 

#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
typedef long long ll;

const int N=1e5+5;
vector<vector<int> >v(N);
int n,cnt=1;
bool flag=true;
vector<int>ans;

void dfs(int n)
{
	ans.push_back(n);
	for(auto i:v[n])
	{
		dfs(i);
	}
}

void output()
{
	for(int i=0;i<ans.size();i++)
	{
		if(i)cout<<' ';
		cout<<ans[i];
	}
}

int main()
{
	int maxn=0;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		int x;cin>>x;
		v[x].push_back(i);
	}
    for(int i=1;i<=n;i++)maxn=max(maxn,(int)v[i].size());
	for(int i=1;i<=n;i++)
	{
		if(v[i].size()!=0 && v[i].size()!=maxn)flag=false;
	}
	cout<<maxn<<' ';if(flag)cout<<"yes";else cout<<"no";cout<<endl;
	dfs(v[0][0]);
	output();
}