【PAT】1138 Postorder Traversal (25 分)

 一、不用map找inRoot

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

vector<int> in,pre;

int flag=0;
void lca(int inl,int inr,int preRoot){
    if(inl>inr || flag) return;
    int i=inl;
    while(in[i]!=pre[preRoot]) i++;
    lca(inl,i-1,preRoot+1);
    lca(i+1,inr,preRoot+(i-inl+1));
    if(flag==0){
        printf("%d",pre[preRoot]);
        flag=1;
    }
}

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]);
    }
    lca(0,n-1,0);
    return 0;
}

二、map 找inRoot

#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;
}