UVA 548 —— 二叉树的递归遍历

已知二叉树的中序遍历和后序遍历，构建二叉树，求叶子到根的最小路径权值和；

后序遍历的最后一个字符即为二叉树的根结点，然后在中序遍历中找到这个根结点的位置，根结点的左边序列即为该根的左子树，右边序列即为右子树，然后再在后序遍历中找左右子树的根结点，以此类推，用递归遍历。

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
const int maxn = 1e4+5;
int in_order[maxn], post_order[maxn];
int lch[maxn], rch[maxn];
int cnt;

bool read_input(int* a){
    cnt = 0;
    string s;
    if(!getline(cin, s))    return false;
    int len = (int)s.size();
    int tmp = 0;
    for(int i = 0;i<len;i++){
        if(s[i] == ' '){
            a[cnt++] = tmp;
            tmp = 0;
            continue;
        }
        tmp = tmp*10+(s[i]-'0');
    }
    a[cnt++] = tmp;
    if(cnt > 0) return true;
    else    return false;
}

int build(int L1, int R1, int L2, int R2){
    if(L1 > R1) return 0;       //空树
    int root = post_order[R2];
    int i;
    for(i = L1;i<=R1;i++){
        if(in_order[i] == root)
            break;
    }
    int k = i-L1;
    lch[root] = build(L1, i-1, L2, L2+k-1);
    rch[root] = build(i+1, R1, L2+k, R2-1);
    return root;
}

int ans, tmp;
void dfs(int u, int sum){
    if(u == 0)  return;
    sum += u;
    if(!lch[u] && !rch[u]){     //叶子
        if(ans > sum || (ans == sum && tmp > u)){
            ans = sum;
            tmp = u;
        }
        return;
    }
    dfs(lch[u], sum);
    dfs(rch[u], sum);
}

int main(){
    while(read_input(in_order)){
        read_input(post_order);
        build(0, cnt-1, 0, cnt-1);
        ans = INF;
        dfs(post_order[cnt-1], 0);
        cout<<tmp<<endl;
    }
    return 0;
}