预备知识:
- 二叉树前序和中序转后序算法
题目翻译:
一个中序遍历的二叉树可以通过堆栈以非递归的方式实现。例如:假设遍历一棵拥有6个结点的二叉树(编号从1到6),对栈的操作为:1入栈,2入栈,3入栈,元素出栈,元素出栈,4入栈,元素出栈,元素出栈,5入栈,6入栈,元素出栈,元素出栈。由此可得到一颗唯一的二叉树(如图1所示)。你的任务就是给出这棵树的后序遍历序列。
难点单词:
暂无
输入格式:
每个输入文件包含一个测试用例。对于每个用例,第一行包含一个正整数N(≤30)代表树的节点总数(编号从1到N)。然后2N行,每个给出了一个对于堆栈的操作。“Push X”代表X入栈,“Pop” 代表栈顶元素出栈。
输出格式:
对于每个测试用例,在一行中打印相应树的后序遍历序列。解决方案肯定存在。所有的数字必须用一个空格分隔,行尾不能有多余的空格。
输入样例:
6
Push 1
Push 2
Push 3
Pop
Pop
Push 4
Pop
Pop
Push 5
Push 6
Pop
Pop
输出样例:
3 4 2 6 5 1
题目解析:
重要知识点:二叉树若用栈实现的中序遍历,每次Push入值的顺序是二叉树的前序遍历。
所以根据题目给出的条件,我们可以知道这个树的前序和中序遍历。此时若要求后序遍历的结果,可以采用
「二叉树前序和中序转后序」的方法。
所以解题目标就变成了:
- 获取二叉树的前序序列;
- 获取二叉树的中序序列;
- 前序中序转后序;
- 输出结果。
由于题目并没有说明所有节点的值互不相同。因此在有多个节点的值相同的情况下,代码可能会出现错误。
所以我们将存储具体值的地方改为存储索引(前中后序中均保存索引值),然后用value数组存储每个索引对应的具体值。
逻辑梳理:
核心变量:pre、in、post中的存储索引值,value存储具体值。
vector<int> pre, in, post, value;
-
获取一个输入,判断该输入的类型:若为Push,存储入栈的值到前序序列(同时存储下标和值),执行入栈操作;若为POP,存储出栈的值到中序序列(同时存储下标和值),执行出栈操作。
//单次输入的情况 char str[5]; if (strlen(str) == 4) { // 输入为Push int num; scanf("%d", &num); value.push_back(num);//存入具体值(v下标->具体值) pre.push_back(key);//存入前序索引(索引值->v下标) s.push(key++);//入栈//中序 } else { // 输入为POP in.push_back(s.top());//记录中序索引(索引值->v下标) s.pop();//出栈//中序 }
-
所有数据(操作)输入完成后,此时pre存储序前结点索引,in中存储序结点索引,value存储索对应的具体值。直接进行二叉树的前序中序转后序。(这里未来会单独开一篇博客讲解)
若对「二叉树前序和中序转后序」不熟悉的同学可以先点击学习一下。//二叉树前中序转后序 // root为当前子树的根结点在前序pre中的下标 // start和end为当前子树最左边和最右边的结点在中序in中的下标 void postorder(int root, int start, int end) { if (start > end) return; int i = start; while (i < end && in[i] != pre[root]) i++; postorder(root + 1, start, i - 1); postorder(root + 1 + i - start, i + 1, end); post.push_back(pre[root]); }
-
转换结束后直接输出后序序列即可
printf("%d", value[post[0]]); for (int i = 1; i < n; i++) printf(" %d",value[post[i]]);
参考代码:
#include <cstdio>
#include <vector>
#include <stack>
#include <cstring>
using namespace std;
// pre、in、post中的存储索引值,value存储具体值
vector<int> pre, in, post, value;
//二叉树前中序转后序
// root为当前子树的根结点在前序pre中的下标
// start和end为当前子树最左边和最右边的结点在中序in中的下标
void postorder(int root, int start, int end) {
if (start > end) return;
int i = start;
while (i < end && in[i] != pre[root]) i++;
postorder(root + 1, start, i - 1);
postorder(root + 1 + i - start, i + 1, end);
post.push_back(pre[root]);
}
int main() {
int n;
scanf("%d", &n);
char str[5];
stack<int> s;
int key=0; //索引(前中后序均保存索引值)
while (~scanf("%s", str)) { // 等价于scanf()!=EOF
if (strlen(str) == 4) { // 输入为Push
int num;
scanf("%d", &num);
value.push_back(num);//存入具体值(v下标->具体值)
pre.push_back(key);//存入前序索引(索引值->v下标)
s.push(key++);//入栈//中序
} else { // 输入为POP
in.push_back(s.top());//记录中序索引(索引值->v下标)
s.pop();//出栈//中序
}
}
//while结束后 此时pre存储序前结点索引 in中存储序结点索引 value存索对应的具体值
postorder(0, 0, n - 1);
printf("%d", value[post[0]]);
for (int i = 1; i < n; i++)
printf(" %d",value[post[i]]);
return 0;
}
全部通过