UVA - 548 Tree 二叉树

来源:互联网 发布:淘宝物品怎么靠前搜到 编辑:程序博客网 时间:2024/05/21 03:17

题目: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;}


0 0
原创粉丝点击