1138 Postorder Traversal (25 分)

这道题的数据稍微有点大，但是题目问的也是比较限制特殊情况，那就是输出后续遍历的第一个节点，后续遍历的特点就是先左后右再根，所以我们可以在递归的时候按照优先级来递归，避免无效递归，如果有左节点，那就向左递归，这时候不管右边有没有都向左递归，如果没有左节点有右节点，那么就向右边递归，如果左右节点都没有，那么就是到头了，输出答案即可，然后还有一个可以优化的地方就是用map来记录中序遍历的位置，最后40ms即可通过，

#include <bits/stdc++.h>

#define fi first
#define se second
#define pb push_back
#define all(x) (x).begin(), (x).end()

using namespace std;

typedef long long ll;
typedef vector<int> vi;
typedef pair<int, int> pa;

const int N = 5e4 + 5;

int pre[N], in[N];
map<int, int> mp;

void solve(int prel, int prer, int inl, int inr) {
    int k = mp[pre[prel]];
    if (k > inl) solve(prel + 1, prel + k - inl, inl, k - 1);
    else if (k < inr) solve(prel + k - inl + 1, prer, k + 1, inr);
    else cout << pre[prel];
}

int main() {
    int n;
    cin >> n;
    for (int i = 0; i < n; i++) scanf("%d", &pre[i]);
    for (int i = 0; i < n; i++) { scanf("%d", &in[i]); mp[in[i]] = i; }
    solve(0, n - 1, 0, n - 1);
    return 0;
}