【PAT】1138 Postorder Traversal (25 分)

#include <iostream>
#include <vector>
#include <map>
using namespace std;

map<int,int> pos;
vector<int> in,pre,post;


void lca(int inl,int inr,int preRoot){
    if(inl>inr) return;
    int inRoot=pos[pre[preRoot]];
    lca(inl,inRoot-1,preRoot+1);
    lca(inRoot+1,inr,preRoot+(inRoot-inl+1));
    post.push_back(pre[preRoot]);
}

int main(){
    int n;
    scanf("%d",&n);
    pre.resize(n);in.resize(n);//resize会重新设置vector的大小，并且元素初始化为零
    for(int i=0;i<n;i++)  scanf("%d",&pre[i]);
    for(int i=0;i<n;i++){
        scanf("%d",&in[i]);
        pos[in[i]]=i;
    }
    lca(0,n-1,0);
    printf("%d",post[0]);
    return 0;
}