uva548--tree(二叉树的递归遍历)

题意:给一棵点带权的二叉树(权职各不相同,都是小于10000的整数)的中序和后序遍历,找一个叶子使得它到根节点的权和最小,如果有多解就输出最小的权和,输入两行,第一行是中序遍历,第二行是后序遍历:

输入:

3 2 1 4 5 7 6

3 1 2 5 6 7 4

7 8 11 3 5 16 12 18

8 3 11 7 16 18 12 5

255 

255

输出:

1

3

255


分析:

后序遍历的最后一个元素就是根节点,而且各结点的权值都不相同,因此我们可以讲权值作为编号储存在中序和后序数组中,所以

只需要在后序遍历中将根节点找出来,然后找中序遍历的第一个元素也就是左子树的最小值,(最左边),然后就能在中序遍历

找到根节点的位置,相应的顺便把左子树的结点个数也得出来,就能知道左子树和右子树的中序,后序遍历了,从而就能找到左右

子树的结点列表,然后就可以把二叉树构建出来了。然后在进行dfs递归,来找到最优解。下面上代码。

<span style="font-size:14px;">#include <bits/stdc++.h>
using namespace std;

const int maxn=10010;
int in_order[maxn],post_order[maxn],n,rch[maxn],lch[maxn];
bool input(int *a) //输入函数
{
    string line;
    if(!getline(cin,line)) //输入字符串时防止空格停止输入
        return false;
    stringstream ss(line); //:stringstream的基本用法stringstream是字符串流,它将流与存储在内存中的string对象绑定起来,在多种数据类型之间实现自动格式化.
    n=0;
    int x;
    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=post_order[R2]; //二叉树的后序遍历最后一个元素是根节点
    int p=L1;
    while(in_order[p]!=root) //中序遍历找到根结点
        p++; // p是根节点的权值
    int cnt=p-L1;//左子节点的个数
    lch[root]=build(L1,p-1,L2,L2+cnt-1); //中序的左子树和后序的左子树
    rch[root]=build(p+1,R1,L2+cnt,R2-1); //中序的右子树和后序的右子树
    return root;
}
int best,best_sum; //目前为止的最优解和最优解的权和
void dfs(int u,int sum) //递归遍历找最优解,u是根节点的权值
{
    sum+=u;
    if(!lch[u]&&!rch[u]) //表示叶子
    {
        if(sum<best_sum||(sum==best_sum&&u<best))
        {
            best=u;
            best_sum=sum;
        }
    }
    if(lch[u])dfs(lch[u],sum); //将二叉树的权值作为编号
    if(rch[u])dfs(rch[u],sum);
}
int main()
{
    while(input(in_order))
    {
        input(post_order);
        build(0,n-1,0,n-1);
        best_sum=100000000000;
        dfs(post_order[n-1],0);
        cout <<best<<endl;
    }
    return 0;
}</span>


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值