L2-3 满树的遍历

L2-3 满树的遍历

分数 25

全屏浏览

切换布局

作者 陈越

单位 浙江大学

一棵“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<iostream>
#include<vector>
using namespace std;

const int N=1e5+8;

vector<int> tree[N],pre;

int maxt,root;
bool hefa=true;

void dfs(int x){
	pre.push_back(x);
	
	if(tree[x].size()==0)return;
	if(tree[x].size()!=maxt)hefa=false;
	for(int i=0;i<tree[x].size();i++){
		dfs(tree[x][i]);
	}
	
}

int main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0); 
	int n;cin>>n;
	for(int i=1;i<=n;i++){
		int fu;cin>>fu;
		if(fu==0){
			root=i;
		
		}else tree[fu].push_back(i);
	} 
	maxt=tree[root].size();
	dfs(root);
	
	cout<<maxt<<" "<<(hefa?"yes\n":"no\n");
	for(int i=0;i<pre.size();i++){
		cout<<pre[i]<<" \n"[i==pre.size()-1];
	}
	
}