Uva548 Tree 二叉树遍历

/*
    题目描述：见紫书155页

    方法：这道题的一个收获就是若给出一棵树的中序遍历和后序遍历的结果，可以还原该树。
              具体还原方法是后序遍历的最后一个节点是该树的根节点，再中序遍历的结果中找到根节点，然后找到根节点
              的左右子树，分别建立左子树和右子树，在建立左子树时需要先找到左子树的根节点， 然后建立左子树的左右
              子树......如此递归建树。建树后用dfs来找一个满足题目要求的最小权值即可。
*/
#pragma warning(disable:4786)
#pragma comment(linker, "/STACK:102400000,102400000")
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<vector>
#include<cmath>
#include<string>
#include<sstream>
#define LL long long
#define FOR(i,f_start,f_end) for(int i=f_start;i<=f_end;++i)
#define mem(a,x) memset(a,x,sizeof(a))
#define lson l,m,x<<1
#define rson m+1,r,x<<1|1
using namespace std;
const int INF = 0x3f3f3f3f;
const int mod = 1e9 + 7;
const double PI = acos(-1.0);
const double eps=1e-8;
const int maxn = 1e4 + 5;
int lch[maxn] , rch[maxn] , value[maxn],inorder[maxn],postorder[maxn];
int minsum , minv , n ,ans;
void init()
{
    mem(lch,0);     mem(rch,0);     mem(value, 0);
    for(int i = 1;i<=n;i++)
        value[i] = inorder[i];
}
/*
    build含义：inorder[L1~R1]与postorder[L2~R2]是一棵树的中序遍历和后序遍历的结果，build函数的功能
                     就是将这些节点建树后再返回这棵树的根节点

    建树后， k号节点的权值是inorder[k]，因此整棵树的根节点的编号不一定是1，而是满足inorder[s] ==postorder[n]
    的整数s

    注意这种递归建树是很常用的建树方法
*/
int build(int L1 , int R1 , int L2 , int R2)
{
    int root ;
    for(root = L1;root<=R1;root++){
        if(inorder[root]==postorder[R2])       break;
    }
    value[root] = postorder[R2];
    int left_length = root - L1 , right_length = R1 - root;
    if(left_length !=0)
        lch[root] = build(L1 , root - 1 , L2 , L2 + left_length -1);
    if(right_length != 0)
        rch[root] = build(root + 1 , R1 , L2 + left_length , R2 - 1 );
    return root;
}
bool read(int * a)          //当输入比较复杂时，记得使用bool型read函数
{
    string line;
    if(!getline(cin , line))        return false;
    stringstream ss(line);
    int x ;
    n = 0  ;
    while(ss >> x)      a[++n] = x;
    return true;
}
void dfs(int root , int sum)
{
    if(lch[root] == 0 && rch[root] == 0){
        if(minsum >= sum || (minsum == sum && minv >= value[root])){
            minsum = sum;
            ans = minv = value[root];
        }
        return ;
    }
    int u = lch[root];
    if(u!=0)
        dfs(u , sum + value[u]);
    int v = rch[root];
    if(v!=0)
        dfs(v , sum + value[v]);
}
int main()
{
    while(read(inorder)){
        read(postorder);
        minsum = INF;      minv = INF;
        init();
        int rroot = build(1 , n , 1 , n);
        dfs(rroot , value[rroot]);
        printf("%d\n",ans);
    }
    return 0;
}