数据结构实验——二叉树的还原和遍历

二叉树的还原和遍历
题目描述
​已知二叉树先序遍历的结果（含空指针记号’#’），可以还原出二叉树。如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;
}