7-38 列出叶结点 (300 分)
对于给定的二叉树,本题要求你按从上到下、从左到右的顺序输出其所有叶节点。
输入格式:
首先第一行给出一个正整数 N(≤10),为树中结点总数。树中的结点从 0 到 N−1 编号。随后 N 行,每行给出一个对应结点左右孩子的编号。如果某个孩子不存在,则在对应位置给出 “-”。编号间以 1 个空格分隔。
输出格式:
在一行中按规定顺序输出叶节点的编号。编号间以 1 个空格分隔,行首尾不得有多余空格。
输入样例:
8
1 -
0 -
2 7
5 -
4 6
输出样例:
4 1 5
存储:本题所给数据方式便于存储父亲编号,但是左右节点都指向父亲节点,便于区分左右,左孩子存储父节点编号的相反数。右孩子直接存储父亲编号,但是0的左右孩子存储±0无区别,故编号1-N合适。
遍历:题目要求从上至下,从左至右,使用quene队列层序遍历,如上存储结果,根节点存储0,找到根压入队列,层序遍历,如果找到没有子节点的节点(没有节点的绝对值指向自己编号)就输出改编号减一即可
数据演示如下
//让编号从1到N,pre存储父节点,如pre[5] = -8 pre[7] = 8表示 8 8->L=5 8->R=7;
#include <bits/stdc++.h>
using namespace std;
int main() {
int n;
char nA, nB;
vector<int> pre(10000, 0);
queue<int> q;
bool falgA = false;
cin >> n;
for (int i = 1; i <= n; i++) {
cin >> nA >> nB;
if(nA!='-'){
pre[(nA-'0') + 1] = -i;//左孩子标记父亲
}
if (nB != '-') {
pre[(nB - '0') + 1] = i;//右孩子标记父亲
}
}
for (int i = 1; i <= n; i++) {//找根
if (!pre[i]) {
q.push(i);
pre[i] = 0;
break;
}
}
while (!q.empty()){
bool falgB = true;//标记该节点为叶节点
for (int i = 1; i <= n; i++) {//找左孩子
if (pre[i] == -q.front()) {
q.push(i);
pre[i] = 0;
falgB = false;
}
}
for (int i = 1; i <= n; i++) {//找右孩子
if (pre[i] == q.front()) {
q.push(i);
pre[i] = 0;
falgB = false;
}
}
if (falgB) {//该节点是叶节点
if (!falgA) {
cout << q.front() - 1;
falgA = true;
}
else {
cout << " " << q.front() - 1;
}
}
q.pop();
}
}