uva548 -- Tree

题目链接

这一题输入二叉树的中序序列和后序序列，让求一个叶子结点使此叶子节点到根节点路径的权值和最小。这题主要有以下几个问题：1、输入方式；2、根据中序和后序序列建立二叉树；3、使用DFS求权值和最小的叶子结点。我是看了刘汝佳的《算法竞赛入门经典》后才做的这题的，所以代码大体上和刘的相同，输入也是采用stringstream，只是在书中刘是采用数组的方式实现的二叉树，在我的代码中我将其改为指针的方式。具体代码如下：

#include <iostream>
#include <cstdio>
#include <string>
#include <sstream>
using namespace std;

struct Node
{
    int value;
    Node* left;
    Node* right;

    Node():value(0),left(NULL),right(NULL){}
};

const int N=10000+10;
int inOrder[N];     //中序序列
int postOrder[N];   //后序序列
int n;  //中序与后序序列的长度
int least;  //最小的权和

bool input(int *a)  //输入函数
{
    string line;
    if(!getline(cin,line))
        return false;
    stringstream ss(line);
    n=0;
    int val;
    while(ss>>val)
        a[n++]=val;
    return n>0;
}

//inOrder[L1,R1],postOrder[L2,R2]
Node* buildTree(int L1,int R1,int L2,int R2)    //建立二叉树
{
    if(L1>R1)
        return NULL;
    Node* root=new Node();
    root->value=postOrder[R2];
    int pos=0;
    for(;pos<n;pos++)
        if(inOrder[pos]==postOrder[R2])
            break;
    int cnt=pos-L1;
    root->left=buildTree(L1,pos-1,L2,L2+cnt-1);
    root->right=buildTree(pos+1,R1,L2+cnt,R2-1);
    return root;
}

void dfs(Node* root,Node*& bestNode,int sum)
{
    if(root==NULL)
        return;
    sum+=root->value;
    if(root->left==NULL&&root->right==NULL)
    {
        if(sum<least||sum==least&&root->value<bestNode->value)
        {
            least=sum;
            bestNode=root;
        }
        return;
    }
    dfs(root->left,bestNode,sum);
    dfs(root->right,bestNode,sum);
}

int main()
{
    //freopen("test.txt","r",stdin);
    while(input(inOrder))
    {
        input(postOrder);
        Node* root=buildTree(0,n-1,0,n-1);
        int sum=0;
        Node* bestNode=root;
        least=100000000;
        dfs(root,bestNode,sum);
        cout<<bestNode->value<<endl;
    }
    return 0;
}

做完这一题后，如果还没做过根据先序序列和后序序列建立二叉树，可以做uva536，来加深一下建立二叉树的操作。我的关于uva536的题解在此。