用先序序列和中序序列创建二叉树

题目描述

输入格式

第一行： 树的中序遍历

第二行： 同样的树的前序遍历

输出格式

单独的一行表示该树的后序遍历。

输入输出样例

输入 #1复制

ABEDFCHG
CBADEFGH 

输出 #1复制

AEFDBHGC

模板如下：重在partition的过程，参数极容易出错

#include <bits/stdc++.h>
using namespace std;

const int MAX = 1000005;
const int INF = 0x7fffffff;
int N;
char pre[MAX];
char in[MAX];
char re[MAX];

struct Node {
    char v;
    struct Node* lChild = nullptr;
    struct Node* rChild = nullptr;
};

Node* partition(int pl, int pr, int il, int ir) {
    if(pl > pr) return nullptr;
    Node* node = new Node();
    node->v = pre[pl];
    int i;
    for (i = il; i < ir; ++i) {
        if(in[i]==pre[pl]) break;
    }
    int len = i - il;
    node->lChild=partition(pl + 1, pl + len, il, i - 1);
    node->rChild=partition(pl + len + 1, pr, i + 1, ir);
    return node;
}

void reOder(Node* node) {
    if(node== nullptr) return;
    reOder(node->lChild);
    reOder(node->rChild);
    printf("%c",node->v);
}

int main() {
    //freopen("../data","r",stdin);
    scanf("%s%s",in,pre);
    Node* head = partition(0,(int)strlen(pre)-1,0,(int)strlen(in)-1);
    reOder(head);
}