输入有多组,每组包含两行,第一行是二叉树的中序遍历,第二行是后序遍历。每个数字代表结点的值,输出某个叶子结点的值,表示从根节点到该叶子节点经过的结点的和最小(包括根节点和叶子结点),如果有多个相同,那么输出叶子结点的值最小的那个结点的值。
思路:根据题目中给出的中序和后序递归建树。dfs遍历所有的情况和,记录最小的情况。
#include<cstdio>
#include<sstream>
#include<iostream>
#include<string>
using namespace std;
const int maxv = 10000+10;
int n;
int in_order[maxv], post_order[maxv], lch[maxv], rch[maxv];
//读取遍历
bool read_list(int* a){
string line;
if(!getline(cin,line))return false;
stringstream ss(line);
int x;
n=0;
while(ss>>x)a[n++]=x;
return n>0;
}
//建树
//in_order[L1...R1] post_order[L2...R2]
int build(int L1,int R1,int L2,int R2){
if(L1>R1)return 0;
int p = L1;
int root = post_order[R2];//根结点
while(root!=in_order[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;
}
//dfs
int best_sum,best;
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(){
// freopen("input.txt","r",stdin);
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<<endl;
}
return 0;
}