题目链接
一棵“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>
#include<algorithm>
using namespace std;
const int N = 10e5 + 1;
vector<int>Tree[N];
int counts = 0;
int n;
void prePrint(int a){
if (a){
if (!counts) cout << a;
else cout << " " << a;
counts ++;
if (counts == n) return ;
}
for (int i = 0; i < Tree[a].size(); i ++){
prePrint(Tree[a][i]);//遍历结点 到第二个结点的时候,如果没有结点就不会进去这个循环
}
}
int main(){
cin >> n;
for (int i = 1; i <= n; i ++){
int a;
cin >> a;
Tree[a].push_back(i);//以父节点 为根据,把字结点全部放进去
}
int max = 0, flog = 0;//max记录最大度
for (int i = 1; i <= n; i ++){
sort(Tree[i].begin(), Tree[i].end());
int length = Tree[i].size();//记录里面的长度,对应它的度
// cout << length;
if (length > max){
max = length;
flog ++;//flog标记有没有不一样的度
}
}
// cout << flog ;
cout << max << " ";
flog > 1 ? cout << "no" << endl : cout << "yes" << endl;
counts = 0;
prePrint(0);
return 0;
}
做自己能力之外的才叫做进步