通过先序和中序数组生成后序数组

通过先序和中序数组生成后序数组

题目描述

给出一棵二叉树的先序和中序数组，通过这两个数组直接生成正确的后序数组。

输入描述：

第一行一个整数 n，表示二叉树的大小。

第二行 n 个整数 a_i，表示二叉树的先序遍历数组。

第三行 n 个整数 b_i，表示二叉树的中序遍历数组。

输出描述：

输出一行 n 个整数表示二叉树的后序遍历数组。

示例1

输入

3
1 2 3
2 1 3

输出

2 3 1

备注：

$1\le n\le 10^4$

$1\le a_i,b_i\le n$

---

题解：

先序遍历：根 左 右

中序遍历：左 根 右

先序遍历的第一个节点一定是整棵树的根节点，于是我们可以在中序遍历中查找该节点位置，假设为 p ，则在中序遍历序列中：p左边的序列就是根节点左子树中序遍历序列，p右边的序列就是根节点右子树中序遍历序列，递归进行处理左右子树即可。(注意：提前存储先序序列中的每个元素在中序序列中的位置，可优化速度)

代码：

#include <cstdio>
#include <unordered_map>

using namespace std;

const int N = 10010;

int n;
int pre[N];
int in[N];
unordered_map<int, int> Hash;

void dfs(int p_l, int p_r, int i_l, int i_r) {
    if (p_l > p_r) return;
    int p = Hash[pre[p_l]];
    int k = p - i_l;
    dfs(p_l + 1, p_l + k, i_l, p - 1);
    dfs(p_l + k + 1, p_r, p + 1, i_r);
    printf("%d ", pre[p_l]);
}

int main(void) {
    scanf("%d", &n);
    for (int i = 0; i < n; ++i) scanf("%d", pre + i);
    for (int i = 0; i < n; ++i) {
        scanf("%d", in + i);
        Hash[in[i]] = i;
    }
    dfs(0, n - 1, 0, n - 1);
    return 0;
}