Tree UVa548

本题要求使用一个二叉树中序遍历和后序遍历的结果,反向构造出树。

看的紫书的代码,建树过程主要通过build函数,传递4个参数,中序遍历起点、终点,后序遍历起点、终点,递归地建出整个树,注意到,后序遍历的最后一个数据为当前子树的根节点,在中序遍历结果中找到该节点,可以得到该节点左边范围为左子树,右边为右子树。此时开left和right两个数组,直接对权值进行编号,数据记录了左右儿子的序号,也就是左右儿子的权值。由此进行递归,将当前节点的左右儿子用build函数赋值,分别递归左子树和右子树,同时函数本身最后会返回当前根节点的权值。递归终止条件为,中序遍历起点大于终点(不知道为什么紫书没写后序遍历的起点终点关系)。

本题紫书代码研究了半天终于想明白了,最后bfs的时候也没搞清楚最后输出的是啥,又调试了好久。。

最后,AC代码:

#include <iostream>
#include <sstream>
#include <cstring>
#include <string>
#define maxn 10010
using namespace std;

int wt1[maxn], wt2[maxn], len, best, bestSum; 
int lch[maxn], rch[maxn];

bool read(int *a)
{
    string x;
    if (!getline(cin, x)) return false;
    stringstream ss(x);
    int i = 0;
    while (ss >> a[i])
    {
        i++;
    }
    len = i - 1;
    return i;
}

int build(int l1, int r1, int l2, int r2)
{
    if (l1 > r1 || l2 > r2)    return 0;
    int root = wt2[r2], i = l1;
    while (wt1[i] != root)  i++;
    int cnt = i - l1;
    lch[root] = build(l1, i - 1, l2, l2 + cnt - 1);
    rch[root] = build(i + 1, r1, l2 + cnt, r2 - 1);
    return root;
}

void dfs(int x, int sum)
{
    sum += x;
    if (!lch[x] && !rch[x])
        if (sum < bestSum || (sum == bestSum && x < best))
        {
            bestSum = sum;
            best = x;
        }
    if (lch[x]) dfs(lch[x], sum);
    if (rch[x]) dfs(rch[x], sum);
}

int main()
{
    while (read(wt1))
    {
        read(wt2);
        build(0, len, 0, len);
        bestSum = 1 << 30;
        dfs(wt2[len], 0);
        cout << best << endl;
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值