题意
Little Valentine非常喜欢玩二叉树。她最喜欢的游戏是用节点中的大写字母构造随机查找的二叉树。
这是她的创作之一的一个例子:
D
/ \
/ \
B E
/ \ \
/ \ \
A C G
/
/
F
为了记录后代的树,她为每棵树写下了两个字符串:遍历(根,左子树,右子树)和有序遍历(左子树,根,右子树)。对于上面绘制的树,预遍历为DBACEGF,有序遍历为ABCDEFG。
她认为这样的一对字符串会提供足够的信息,以便以后重建树(但她从未尝试过)。
现在,几年后,她再次看了看琴弦,意识到重建树木确实是可能的,但这仅仅是因为她从未在同一棵树上两次使用过相同的字母。
但是,手工重建很快变得很乏味。
因此,现在她要求您编写一个可以为她完成工作的程序!
输入
输入将包含一个或多个测试用例。
每个测试用例由一行包含两个字符串preord和inord组成,分别表示二叉树的预遍历和有序遍历。 两个字符串均由唯一的大写字母组成。 (因此,它们不能超过26个字符。)
输入在文件末尾终止。
输出
对于每个测试用例,恢复Valentine的二叉树并打印包含该树的后序遍历的一行(左子树,右子树,根)。
样例
Sample Input
DBACEGF ABCDEFG
BCAD CBAD
Sample Output
ACBFGED
CDAB
解题思路
通过前序遍历的首个字母,我们可确定树的根节点,同时,利用该根节点在中序遍历的位置,我们能够将左右子树的中序遍历列出,那么我们只要继续按照前序遍历的遍历方式,不断找出每一个子树的根节点以及左右子树的中序遍历序列,即可列出整棵树。
而后,由于题目要求输出树的后序遍历,那么只要在每一次递归时,寻找左右子树完毕后,将当前树的根节点输出,即可。
AC代码
#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
string preorder;//前序遍历
void postorder(string inorder) {
if (inorder.length() == 0) { //当中序遍历为空说明没有子节点
return;
}
char a = preorder[0];//当前树的根节点
preorder = preorder.substr(1);//选出当前树根结点后,前序遍历的首位应当去掉
int pos = 0;//寻找当前树根节点位于中序遍历的位置
for (int i = 0; i < inorder.length(); i++) {
if (a == inorder[i]) {
pos = i;
break;
}
}
string inorderLeft = inorder.substr(0, pos);//当前树的左子树的中序遍历
string inorderRight = inorder.substr(pos + 1);//当前树的右子树的中序遍历
postorder(inorderLeft);//遍历左子树
postorder(inorderRight);//遍历右子树
cout << a;
}
int main() {
string inorder;//中序遍历
while (cin >> preorder >> inorder) {
postorder(inorder);
cout << endl;
}
return 0;
}