P1030 求先序排列
题目大意:
给定一棵二叉树的中序和后序遍历,求先序遍历。
例如中序遍历\(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;
}