Tree UVA - 548

不得不说这个题递归用的非常的巧妙,仍然是仿照的刘汝佳的代码;

大致讲一下思路:

这是第一个样例

3 2 1 4 5 7 6

3 1 2 5 6 7 4

后序遍历最后一个是根节点,4,从中序遍历里面找到4是在中间,由此中序遍历就被分为 左子树3 2 1和右子树 5 7 6

然后后序遍历也被分为3 1 2 和 5 6 7 由此开始重复递归就完成了建树的过程

求权和的过程用了深搜个人认为这个还是比较好理解的;

贴上刘汝佳的代码

#include <iostream>
#include <stdio.h>
#include <string>
#include <sstream>
#include <algorithm>
using namespace std;
const int maxv=10000+10;
int in_order[maxv],post_order[maxv],lch[maxv],rch[maxv];
int n;
bool read_list (int *a)
{
    string line;
    if(!getline(cin,line))
        return false;
    stringstream ss(line);
    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++;
    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)
{
    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(read_list(in_order))
    {
        read_list(post_order);
        build(0,n-1,0,n-1);
        best_sum=1000000;
        dfs(post_order[n-1],0);
        cout<<best<<"\n";
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/W349652743/article/details/80398962
今日推荐