题目描述
时间限制:10000ms
单点时限:1000ms
内存限制:256MB
描述
在参与过了美食节之后,小Hi和小Ho在别的地方又玩耍了一阵子,在这个过程中,小Ho得到了一个非常有意思的玩具——一棵由小球和木棍连接起来的二叉树!小Ho对这棵二叉树爱不释手,于是给它的每一个节点都标记了一个标号——一个属于A..Z的大写字母,并且没有任意两个节点的标号是一样的。小Hi也瞅准了这个机会,重新巩固了一下小Ho关于二叉树遍历的基础知识~就这样,日子安稳的过了两天。这天,小Ho正好在求解这棵二叉树的前序、中序和后序遍历的结果,但是却在求出前序遍历和中序遍历之后不小心把二叉树摔到了地上,小球和木棍等零件散落了一地!小Ho损失了心爱的玩具,正要嚎啕大哭起来,所幸被小Hi发现了,劝说道:“别着急,这不是零件都还在么?拼起来不就是了?”“可是我忘记了二叉树长什么样子了!”小Ho沮丧道。“这个简单,你不是刚刚求出了这棵二叉树的前序和中序遍历的结果么,利用这两个信息就可以还原出整棵二叉树来哦!”这样么?!!”小Ho止住了泪水,问道:“那要怎么做呢?”
没错!小Ho在这一周遇到的问题便是:给出一棵二叉树的前序和中序遍历的结果,还原这棵二叉树并输出其后序遍历的结果。
输入
每个测试点(输入文件)有且仅有一组测试数据。
每组测试数据的第一行为一个由大写英文字母组成的字符串,表示该二叉树的前序遍历的结果。
每组测试数据的第二行为一个由大写英文字母组成的字符串,表示该二叉树的中序遍历的结果。
对于100%的数据,满足二叉树的节点数小于等于26。
输出
对于每组测试数据,输出一个由大写英文字母组成的字符串,表示还原出的二叉树的后序遍历的结果。
样例输入
AB
BA
样例输出
BA
题意—-其实题目那么长主要意思就是给定二叉树的前序和中序,输出它的后序。
思路
- 前序的第一个字符一定是根节点,找到之后再在中序中找到它
- 将中序分为左右两棵子树
- 重复以上两步,即可构造出二叉树
- 递归得到二叉树的后序遍历结果
代码写的很丑,其中有用到string中的find()函数以及substr()函数,具体用法参考我的有关这个的博客,之后想对这个代码进行改进,写出更简洁的代码。
代码如下:
#include<iostream>
#include<string>
using namespace std;
struct Node {
char data;
Node * left;
Node * right;
Node(char a){
data = a;
left=NULL;
right = NULL;
}
};
Node* getNode(string front,string center) {
Node* root = new Node(front[0]);
int len = center.length();
if(len == 1){
return root;
}
int index = center.find(front[0]);
if(index==0 && index !=len-1 ) {
int temp= len-index-1;
string subcenter2 = center.substr(index+1,temp);
string subfront2 = front.substr(1+index,temp);
root->right = getNode(subfront2,subcenter2);
} else if(index!=0 && index ==len-1) {
int temp = index;
string subcenter1 = center.substr(0,temp);
string subfront1 = front.substr(1,temp);
root->left = getNode(subfront1,subcenter1);
}else{
int temp1 = index;
int temp2 = len- index-1;
string subcenter2 = center.substr(index+1,temp2);
string subfront2 = front.substr(1+index,temp2);
string subcenter1 = center.substr(0,temp1);
string subfront1 = front.substr(1,temp1);
root->left = getNode(subfront1,subcenter1);
root->right = getNode(subfront2,subcenter2);
}
return root;
}
void printBack(Node *root) {
if(root->left!=NULL) {
printBack(root->left);
}
if(root->right!=NULL) {
printBack(root->right);
}
cout<<root->data;
return;
}
int main() {
string front,center;
cin>>front>>center;
Node * tree = getNode(front,center);
printBack(tree);
return 0;
}