1086 Tree Traversals Again (25 分) 全网最细 题目详解 翻译 完整代码 PAT甲级真题解析

预备知识：

1. 二叉树前序和中序转后序算法

题目翻译：

一个中序遍历的二叉树可以通过堆栈以非递归的方式实现。例如：假设遍历一棵拥有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入值的顺序是二叉树的前序遍历。

所以根据题目给出的条件，我们可以知道这个树的前序和中序遍历。此时若要求后序遍历的结果，可以采用
「二叉树前序和中序转后序」的方法。

所以解题目标就变成了：

1. 获取二叉树的前序序列；
2. 获取二叉树的中序序列；
3. 前序中序转后序；
4. 输出结果。

由于题目并没有说明所有节点的值互不相同。因此在有多个节点的值相同的情况下，代码可能会出现错误。
所以我们将存储具体值的地方改为存储索引（前中后序中均保存索引值），然后用value数组存储每个索引对应的具体值。

逻辑梳理：

核心变量：pre、in、post中的存储索引值，value存储具体值。

vector<int> pre, in, post, value;

1. 获取一个输入，判断该输入的类型：若为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();//出栈//中序
   }
   

2. 所有数据（操作）输入完成后，此时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]);
   }
   

3. 转换结束后直接输出后序序列即可

   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;
}

全部通过