题目链接:http://hihocoder.com/problemset/problem/1049
题意:输入有两个字符串,代表给出一棵二叉树的前序和中序遍历的结果,让求这后序遍历的结果。
提示如下:
“这可就要从头说起了,我们先找一棵二叉树作为例子吧!”小Hi在纸上画了画,递给了小Ho。
“那么对于这棵二叉树,你先来计算一下它的前序遍历和中序遍历的结果吧!”小Hi也是毫不含糊就开始下发任务。
“唔……前序遍历是ABDEGHCFIJ,中序遍历……我想想啊……是DBGEHACIJF!”小Ho并没有花费多长时间就给出了答案。
“你还记得前序遍历的递归定义么?”小Hi点了点头又继续问道。
“是‘根节点’+‘左子树的前序遍历’+‘右子树的前序遍历’!”小Ho毫不犹豫的答道:“而在这里体现出来就是‘A'+‘BDEGH’+‘CFIJ’”。
“那中序遍历呢?”小Hi继续问道。
“是‘左子树的中序遍历’+‘根节点’+‘右子树的中序遍历’!在这里也就是‘DBGEH’+‘A’+‘CIJF’。”小Ho这次花的时间更少了。
“还记得动态规划时候我们一般处理的方法么?我们这里也可以这样做哦,如果我们定义post_order(str1, str2)为一棵前序遍历的结果为str1,中序遍历的结果为str2的二叉树的后序遍历的结果的话,我们有没有办法把它化解成为子问题呢?”小Hi道。
“让我想想……”小Ho拿出纸笔开始演算起来:“如果我要求解post-order(str1, str2)的话,首先不难发现,根据‘前序遍历’str1=‘根节点’+‘左子树的前序遍历’+‘右子树的前序遍历’,我可以知道这棵二叉树的根节点root便是str1的第一个字符!”小Ho道。
“而我在知道了‘根节点’root之后,我便可以利用‘中序遍历’str2=‘左子树的中序遍历’+‘根节点’+‘右子树的中序遍历’,求解出‘左子树的中序遍历’str2L和‘右子树的中序遍历’str2R!”
“接下来,由于一棵子树的前序遍历和中序遍历的长度相同,那么仍然是根据‘前序遍历’str1=‘根节点’+‘左子树的前序遍历’+‘右子树的前序遍历’,我可以知道从str1的第2个字符开始的str2L.length()个字符便是‘左子树的前序遍历’str1L,而这之后的部分便是‘右子树的前序遍历’str1R!”小Ho说到这里,顿了顿,希望从小Hi处得到些反馈。
“没错!那你准备如何求解post_order(str1, str2)呢?”小Hi肯定道。
“这只要根据之前求出的结果,和‘后序遍历’=‘左子树的后序遍历’+‘右子树的后序遍历’+‘根节点’,便可以知道——post_order(str1, str2)=post_order(str1l, str2l)+post_order(str1r, str2r)+root这样一个式子来!而这个问题的规模——也就是二叉树的大小,是一直在缩小的,所以是能够这样不断的划分做下去的!也就是说我之后可以将当前根节点的左子树又拆分成为两个问题来解决,一直到当前节点是叶子节点了为止!然后慢慢的将这些答案拼成上层的问题的答案,而这个过程只需要使用递归完成就可以了!”
“听起来很容易的样子呢!那你要不要赶紧去实现了算法,算出你的宝贝二叉树长什么样呢?”小Hi满意的点了点头,随即笑着问道。
“那是自然!”
代码:
#include <iostream>
#include <cstdio>
#include <string>
using namespace std;
//定义post_order(str1, str2)为一棵前序遍历的结果为str1,中序遍历的结果为str2的二叉树的后序遍历的结果
//post_order(str1, str2)=post_order(str1l, str2l)+post_order(str1r, str2r)+root
string post_order(string fir,string mid)
{
if(fir.length()==0)
return "";
if(fir.length()==1)
{
string ans="";
ans+=fir[0];
//cout<<"ans="<<ans<<endl;
return ans;
}
char root=fir[0];
//cout<<"root="<<root<<endl;
string firL="",firR="",midL="",midR="";
int it;
for(it=0;it<mid.length();it++)
{
if(mid[it]!=root)
{
midL+=mid[it];
}else{
break;
}
}
for(it=it+1;it<mid.length();it++)
{
midR+=mid[it];
}
for(it=1;it<=midL.length();it++)
{
firL+=fir[it];
}
for(it=midL.length()+1;it<fir.length();it++)
{
firR+=fir[it];
}
return post_order(firL, midL)+post_order(firR, midR)+root;
}
int main()
{
string fir,mid,las;
cin>>fir;
cin>>mid;
las=post_order(fir,mid);
cout<<las<<endl;
return 0;
}