Tree UVA - 548

UVA-548

题目：给一棵点带权（权值各不相同，都是小于10000的正整数）的二叉树的中序和后序遍历，找一个叶子使得它到根的路径上的权和最小。如果有多解，该叶子本身的权应尽量小。输入中每两行表示一棵树，其中第一行为中序遍历，第二行为后序遍历。(题目翻译取自刘汝佳《算法竞赛入门经典》P155)。

#include<cstdio>
#include<cmath>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<vector>
#include<sstream>
using namespace std;
const int MAXN = 20000;
int inOrder[MAXN],postOrder[MAXN],l_tree[MAXN],r_tree[MAXN];
int best,best_leave,n;
bool input(int *a){
    string s;
    while(!getline(cin,s)) return false;
    stringstream ss(s);
    int x;
    n = 0;
    while(ss>>x) a[n++] = x;
    return n>0;
}
int build(int L1,int R1,int L2,int R2){
    if(L1>R1) return 0; //为空树
    int root = postOrder[R2]; 
    int p = L1;
    while(inOrder[p]!=root) p++;
    int count = p - L1; //根坐标为root的左子树的节点个数
    l_tree[root] = build(L1, p-1, L2, L2+count-1); //l_tree[root]指 根坐标为root的左子树的根坐标的值
    r_tree[root] = build(p+1, R1, L2+count, R2-1); //r_tree[root]指 根坐标为root的右子树的根坐标的值
    return root;
}
void dfs(int u,int sum){
    sum += u;
    if(!l_tree[u]&&!r_tree[u]){ //根坐标为u的节点没有左子树也没有右子树，则u是一片叶子
        if(sum<best||(sum==best&&u<best_leave)){
            best = sum; //到目前为止的最优解的对应权值和
            best_leave = u;
        }
    }
    if(l_tree[u]) dfs(l_tree[u],sum); 
    if(r_tree[u]) dfs(r_tree[u],sum);
}
int main()
{
    while(input(inOrder)){
        input(postOrder);
        build(0,n-1,0,n-1);
        best = 1<<30; //best取很大数
        dfs(postOrder[n-1],0);
        cout<<best_leave<<endl;
    }
    return 0;
}

思路：后序遍历的最后一个字符是根，因此只需在中序遍历中找到它，就知道左右子树的中序和后序遍历了。从而可以据此构造二叉树。最后递归遍历，求出最优解。