uva548 tree

#include<iostream>
#include<string>
#include<sstream>
using namespace std;
const int maxv=10000+10;
int a1[maxv],a2[maxv],ltree[maxv],rtree[maxv];
int best=0,best_sum;
/*
题目目的:
 寻找带权最短路径和这个路径上的终端节点 
1.数据的输入
通过streamstring流来进行数据的输入（把空格给消掉了）注意buf应该定义为int型，不能定义为char型，
否则会超时，即要选择什么类型的元素就要用什么类型的buf 
2.利用中序和后序遍历来建树，注意i和cnt值的变化（前序和中序可以类比这个步骤） 
步骤:
a.利用后序遍历找到根节点
b.再在中序遍历中查找根节点的位置
c.利用递归的方法，将中序和后序部分分成两个部分，继续进行查找
d.重复a,b,c步骤直至全部遍历为止 
3.利用DFS算法来查找遍历这个图，寻找带权最短路径和这个路径上的终端节点,也是利用递归来做 
*/


int process(string s,int a[])
{
int i=0;
int buf;
stringstream ss(s);
while(ss>>buf)
{
//cout<<buf<<endl;
a[i++]=buf;
}
return i;
}
//根据后序和中序构建二叉树，核心函数  
int buildTree(int nStartIn,int nEndIn,int nStartPost, int nEndPost)
{  
//cout<<"hrer"<<endl;
    if (nStartIn > nEndIn)
        return 0;  
    int root = a2[nEndPost];  
    int i=nStartIn;  
    while(a1[i]!=root) i++;
    int cnt=i-nStartIn;
    ltree[root] = buildTree(nStartIn, i - 1, nStartPost, nStartPost + cnt - 1);  
    rtree[root] = buildTree(i + 1, nEndIn, nStartPost + cnt, nEndPost - 1);
//cout<<"root="<<root<<endl;  
    return root;  
}
void dfs(int u,int sum)
{
sum+=u;
if(ltree[u]==0&&rtree[u]==0)//注意这里是非空 
{
if(sum<best_sum||(sum==best_sum&&u<best))
{
best_sum=sum;best=u;
}
}
// if(u<minn)
// minn=u;
if(ltree[u])
{
//cout<<"ltree="<<ltree[u]<<endl;
dfs(ltree[u],sum);
}

if(rtree[u])
{
//cout<<"rtree="<<rtree[u]<<endl;
dfs(rtree[u],sum);
}

}
int main()
{
string s1,s2;
//freopen("tree.txt","r",stdin);
while(getline(cin,s1))
{
//数据读入以及存储 
getline(cin,s2);
//cout<<s1<<endl<<s2<<endl;
int n;

n=process(s1,a1);
n=process(s2,a2);
//cout<<"n="<<n<<endl;
//cout<<a1[0]<<a2[0]<<endl;
//根据中序和后序遍历建树的过程 
buildTree(0,n-1,0,n-1);
//cout<<"hrer"<<endl;
//cout<<ltree[a2[n-1]]<<rtree[a2[n-1]]<<endl;
best_sum=10000000;
dfs(a2[n-1],0); 
cout<<best<<endl;
}
return 0;
}