UVA - 548 Tree 二叉树

题目：http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=19105

题意：给定一棵二叉树的中序和后序序列，求从根节点到所有叶子节点的最短路径上的叶子节点，最短路径是指所经过节点的权值和最小，若有多个最短路径，那么取其中叶子节点的最小值

思路：首先构造二叉树。根据对于给定的后序序列，最后一个值肯定是二叉树的根节点，找到中序序列中对应的值的位置，于是可以把中序序列分成两部分，左边是节点的左子树，右边是节点的右子树。然后取后序倒数第二个，如果根节点有右子树，那么此点一定是右子树的根节点，可以把对应的部分再分成左右两部分，以此类推，最后便可构造出二叉树。求最短路径时，直接从根节点开始dfs，加上对应节点的值，搜到叶子节点时去记录最短路径的叶子节点即可

总结：关键就是如何构造二叉树，还有题目输入挺恶心

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;

const int N = 10010;
struct node
{
    int v;
    node *lc, *rc;
    node(int v = 0, node *lc = NULL, node *rc = NULL):v(v),lc(lc),rc(rc){}
};
int arr[N], brr[N];
int ans, res;

void dfs1(int l, int r, int &t, node* &root) //构造二叉树
{
    int flag = -1;
    for(int i = l; i <= r; i++)
        if(arr[i] == brr[t])
        {
            flag = i; break;
        }
    if(flag == -1) return;
    root = new node(arr[flag]);
    t--;
    if(flag < r) //右子树不为空，递归进入右子树
        dfs1(flag + 1, r, t, root -> rc);
    if(flag > l) //左子树不为空，递归进入左子树
        dfs1(l, flag - 1, t, root -> lc);
}
void dfs2(node * r, int tmp)
{
    if(r ->lc != NULL)
        dfs2(r ->lc, r ->lc ->v + tmp);
    if(r ->rc != NULL)
        dfs2(r ->rc, r ->rc ->v + tmp);
    if(r ->lc == NULL && r ->rc == NULL) //走到叶子节点，记录最短路径的叶子节点
    {
        if(ans > tmp)
            ans = tmp, res = r ->v;
        else if(ans == tmp)
            res = min(res, r ->v);
    }
}
int main()
{
    char ch;
    while(~ scanf("%d%c", &arr[0], &ch))
    {
        int n = 1;
        for(int i = 1; ch != '\n'; i++)
            scanf("%d%c", arr + i, &ch), n++;
        ch = ' ';
        for(int i = 0; ch != '\n'; i++)
            scanf("%d%c", brr + i, &ch);
        node *root;
        int t = n - 1;
        dfs1(0, n - 1, t, root);
        ans = res = 0x3f3f3f3f;
        dfs2(root, root ->v);
        printf("%d\n", res);
    }
    return 0;
}