题目大意:
给定一棵二叉树的中序和后序遍历,求先序遍历。
例如中序遍历(ACGDBHZKX),后序遍历为(CDGAHXKZB),那么先序遍历就是(BAGCDZHKX)
思路:
首先我们知道二叉树的先序遍历是先访问根再访问子树,那么对于这道题来说要求先序遍历结果,我们就每次执行函数输出当前子树它的根即可,那么问题来了:
1.根怎么求?
根就是后序遍历的最后一个结点,例如(CDGAHXKZB),则这颗二叉树的根就是(B)。
2.左右子树怎么表示呢?
我们已经找到了根为(B),用(find)函数找到(B)的位置,根据中序遍历是先访问左(右)子树在访问根,然后再访问右(左)子树,可知根就把一颗树分为了左右子树,那么就明白了,对于(ACGDBHZKX),左子树就就是(ACGD),右子树是(HZKX),然后再找后序遍历(CDGA)和(HXKZ)。
因此剩下的我们只需要递归执行同样的过程就可了。
PS:想出这个思路的人一定是个巨佬
代码
#include <iostream>
#include <string>
using namespace std;
//后序遍历找根 中序遍历找根分为两棵子树
void res(string in, string after)
{
if(in.size() > 0)
{
char ch = after[after.size()-1];//找到根并输出
cout << ch;
int k = in.find(ch); //4
res(in.substr(0, k), after.substr(0, k)); //中序遍历左 后序遍历左
res(in.substr(k+1), after.substr(k, in.size()-k-1)); //中序遍历右 后序遍历右
}
}
int main()
{
string in, after;
cin >> in;
cin >> after;
res(in, after);
return 0;
}