满树的遍历
题目难度: ☀️
题目描述:
一棵“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;
}