二叉树的还原和遍历
题目描述
已知二叉树先序遍历的结果(含空指针记号’#’),可以还原出二叉树。如A#B##可以还原出:
A
\
B
已知二叉树先序和中序遍历的结果(不需要空指针记号),也可以还原出二叉树,如先序遍历为ABDC、中序遍历为BDAC,可以还原出:
A
/
B C
\
D
你的任务是判断输入类型,计算还原二叉树(不需要打印),然后输出其后序遍历结点的顺序
输入描述
一个测试用例仅有一行。输入有两种类型,一种是只有先序序列(含空指针记号’#’),一种是有先序序列和中序序列,中间用逗号分隔。
测试用例的结点符号从ASCII字符’A’开始,数量最大达到62个结点。
输出描述
输出后序遍历顺序
提示
可以通过每行最后一个字符是否为’#'来区分两种类型的测试用例。
样例输入
A#B##
样例输出
BA
允许最长运行时间
1000ms
允许使用最大内存
256MB
oj通过100%
代码如下:
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<memory>
#include<iostream>
using namespace std;
struct BinTree
{
char data;
BinTree* lchild, * rchild;
BinTree(char d)
{
data = d;
lchild = NULL;
rchild = NULL;
}
~BinTree()
{
if (lchild)delete lchild;
if (rchild)delete rchild;
}
static BinTree* create(string str, int& index)
{
char ch = str[index++];
if (ch == '#')return NULL;
BinTree* node = new BinTree(ch);
node->data = ch;
node->lchild = create(str, index);
node->rchild = create(str, index);
return node;
}
void PreOrder()
{
printf("%c", data);
if (lchild)lchild->PreOrder();
if (rchild)rchild->PreOrder();
}
void InOrder()//实现中序遍历
{
if (lchild)lchild->InOrder();
printf("%c", data);
if (rchild)rchild->InOrder();
}
void PostOrder()//实现后序遍历
{
if (lchild)lchild->PostOrder();
if (rchild)rchild->PostOrder();
printf("%c", data);
}
static BinTree* rebuild(string PreOrder, string InOrder, int len)
{
if (len == 0)return NULL;
char ch = PreOrder[0];
BinTree* R = new BinTree(ch);
R->data = ch;
int pos = 0;
while (pos < len)
{
if (InOrder[pos] == ch)break;
pos++;
}
R->lchild = rebuild(PreOrder.substr(1, pos), InOrder.substr(0, pos), pos);
R->rchild = rebuild(PreOrder.substr(pos + 1, len - pos - 1), InOrder.substr(pos + 1, len - pos - 1), len - pos - 1);
return R;
}
};
int main()
{
BinTree* tree;
int index = 0;
string data;
string str1;
string str2;
int count = 0;//记录输入字符个数 \n也算入了
char ch;
int i;
cin >> data;
count = data.size();
//printf("%d\n", count);//字符串真实长度
if (data[count - 1] == '#')//判断
{
tree = BinTree::create(data, index);
tree->PostOrder();
}
else
{
int pos = data.find(',');
str1 = data.substr(0, pos);
str2 = data.substr(pos + 1, count - 1);
tree = BinTree::rebuild(str1, str2, pos);
tree->PostOrder();
}
return 0;
}