题目链接:http://acm.tju.edu.cn/toj/showp3988.html
很裸的一道二叉树的题目,题意很简单,通过先序遍历(root,left,right)和中序遍历(left,root,right),推出后序遍历(left,right,root)。
二叉树就是一个搜索的教科书,我们用到的方法也是深度搜索(DFS),举个例子
先序遍历:DBACEFHGI。
中序遍历:ABCDFEGHI。
首先先序遍历中的第一个字母是D,那么D一定是整个二叉树的根节点,那么在中序遍历中D左边的就是左子树,在D右边的就是右子树。
现在我们再来看左子树:
先序遍历:BAC
中序遍历:ABC
同样的方法,先序遍历的第一个B为根节点,左子树为A,右子树为C,那么右子树也是用到同样的方法就能求解。
所以我们得到这个二叉树为
那么我们的答案就是ACBFGIHED。(图是小编自己用软件画的,希望大家支持)
所以我们编写程序的方法也是一样,在每一个数中,先由先序遍历确定根节点,再通过中序遍历确定左子树和右子树。这个过程我们可以用递归去实现。
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
char in[30], pr[30];
void dfs(char *inorder, char *preorder, int len)
{
if(len == 0) return ;
int rootIndex = 0;
for(;rootIndex<len; rootIndex++)
if(inorder[rootIndex] == *preorder) break;
dfs(inorder, preorder+1, rootIndex);
dfs(inorder+rootIndex+1, preorder+rootIndex+1, len-rootIndex-1);
cout << *preorder;
}
int main()
{
while(scanf("%s%s", pr,in) !=EOF)
{
dfs(in, pr, strlen(in));
puts("");
}
}