UVa 548 Tree【二叉树】

题目大意

输入二叉树的中序和后序，找出从叶子节点到根最小的权值和，并输出该叶子节点的值。

解题思路

首先要根据二叉树的中序遍历和后序遍历还原二叉树。

后序遍历的最后一个数，就是根节点的值。可以在中序遍历中找到根节点，根节点左边即为左子树，右边即为右子树，这样就可以还原二叉树了。

而要找到权值和最小的路径，只需要遍历每一个叶子节点，求权值和就可以啦

复杂度分析

递归建树时间复杂度为O(N)，遍历叶子节点时间复杂度为O(N)

心得体会

主要考察的就是中序遍历后序遍历的特点。

再有就是stoi函数，可以把数字字符串转换成int输出，参数为string。atoi函数作用差不多，只是参数为char.

代码

#include<iostream>
#include<algorithm>
#include<vector>
#include<cstring>
#include<sstream>
using namespace std;

struct Node{
    int v;
    Node *left,*right;
    Node():left(NULL),right(NULL){};
};


vector<int> middle,post;                    //中序遍历和后序遍历

Node* creat(int im,int jm,int ip,int jp){   //通过中序和后序建树
    if(im>=jm||ip>=jp)return NULL;          //树为空树
    Node* root=new Node;
    root->v=post[jp-1];                      //后序遍历的最后一个是二叉树的根节点
    int j=find(middle.begin()+im,middle.begin()+jm,post[jp-1]) - middle.begin(); //在中序遍历中查找根节点
    root->left=creat(im,j,ip,ip+(j-im));     //建立左子树
    root->right=creat(j+1,jm,ip+(j-im),jp-1);//建立右子树
    return root;
}

int ans=-1,mins=10e6;

void dfs(Node* nod,int sum){   //计算每个叶子节点的权值和，并记录最小的叶子节点
    if(nod->left==NULL&&nod->right==NULL){ //判断是否为叶子节点
        sum+=nod->v; 
        if(sum<mins){
            mins=sum;
            ans=nod->v;
        }
        else if(sum==mins&&ans>nod->v){ //如果权值和相同，就取叶子的值小的那个
            mins=sum;
            ans=nod->v;
        }
        return;
    }
    if(nod->left!=NULL)dfs(nod->left,nod->v+sum); //找叶子节点
    if(nod->right!=NULL)dfs(nod->right,nod->v+sum);
}

int main()
{
    string s1,s2;
    while(getline(cin,s1) && getline(cin,s2)){ //读取数据
        middle.clear(); //中序
        stringstream ss1(s1);
        string in;
        while(ss1>>in)middle.push_back(stoi(in));//stoi:把数字字符串转换为int值
        post.clear();   //后序
        stringstream ss2(s2);
        while(ss2>>in)post.push_back(stoi(in));
        Node *tree=creat(0,middle.size(),0,post.size()); //建树
        mins=10e6;
        dfs(tree,0);
        cout<<ans<<endl;
    }
    return 0;
}