pat1138 Postorder Traversal

题意：给出二叉树的先序，中序，输出后序的第一个元素。

思路：普通的还原二叉树。

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>

using namespace std;

const int MAX_N = 50010;

int preorder[MAX_N], inorder[MAX_N];
int n;
bool flag;

void post(int l1, int r1, int l2, int r2) {
    if (flag) return ;
    if (l1 >= r1) return ;
    if (r1-l1==1) {
        if (!flag) {
            flag = true;
            printf("%d\n", preorder[l1]); return ;
        }
        return ;
    }
    int p = r2;
    for (int i = l2; i < r2; i++) {
        if (preorder[l1] == inorder[i]) {
            p = i; break;
        }
    }
    post(l1+1, l1+1+p-l2, l2, p);
    post(l1+1+p-l2, r1, p+1, r2);
}

int main() {
    // freopen("in.txt", "r", stdin);
    // freopen("out.txt", "w", stdout);
    scanf("%d", &n); flag = false;
    for (int i = 0; i < n; i++) scanf("%d", &preorder[i]);
    for (int i = 0; i < n; i++) scanf("%d", &inorder[i]);
    post(0, n, 0, n);
    return 0;
}