【二叉树】已知前序、中序求后序

【二叉树】已知前序、中序求后序

代码

#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
vector<int> pre, in, post;
// 后序 左右根
void post_order(int root, int start, int end){
    if(start > end) return; // 递归结束条件
    int i = start;
    while (i < end && in[i] != pre[root]) i++;
    // 从中序里寻找当前根结点的位置i
    post_order(root+1, start, i-1);
    // 左子树递归
    // 左子树根结点为root+1
    // 左子树的中序序列起始点为start
    // 左子树的中序序列结束点为i-1
    post_order(root+i-start+1, i+1, end);
    // 右子树递归
    // 右子树根结点为[root下标+左子树结点个数+1]即[root+(i-start)+1]
    // 右子树的中序序列起始点为i+1
    // 右子树的中序序列结束点为end
    post.push_back(pre[root]);
    // 根结点的存储
}
int main(){
    int n, num;
    // 总结点个数
    scanf("%d", &n);
    for (int i = 0; i < n; i++){
        scanf("%d", &num);
        // 存储先序序列
        pre.push_back(num);
    }
    for (int i = 0; i < n; i++){
        scanf("%d", &num);
        // 存储中序序列
        in.push_back(num);
    }
    // 初始root位置为先序的0位置 初始树的中序起始点为0 结束点为n-1
    post_order(0, 0, n-1);
    // 格式化输出
    printf("%d", post[0]);
    for (int i = 1; i < n; i++)
        printf(" %d", post[i]);
    return 0;
}