数据结构实验 已经先序遍历和中序遍历,构造一个二叉树

给出前序遍历和中序遍历,构造一棵二叉树

/*
给出前序遍历和中序遍历,构造一棵二叉树
明显,由前序遍历和中序遍历可以唯一确定一棵二叉树,

思路分析:前序遍历的第一个数是二叉树的根,在对应的中序遍历中找到这个数,
则他左边的所有数是根的左子树,右边的所有数是根的右子树,
且左边的数的顺序是左子树的中序遍历,右边是右子树的中序遍历,
按照这两个长度,可以在前序遍历中找到对应的左子树的前序遍历和右子树的前序遍历
然后递归调用即可完成.


PS:如果只是单纯由前序遍历和中序遍历推出后序遍历,可以结合栈的使用完成该操作,eg.hdu1710
*/
#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <stack>
using namespace std;
const int maxn = 20;
char pre[maxn];
char in[maxn];

typedef struct Node {
	struct line;
	char val;
	Node *leftchild, *rightchild;
	Node() {}
	Node(char x) {
		val = x;
		leftchild = rightchild = NULL;
	}
}Node;

void build(Node *&root, string pre, string in)
{
	int index = in.find(pre[0]);
	string in_left = in.substr(0, index);//左孩子的中序
	string in_right = in.substr(index + 1);//右孩子的中序
	string pre_left = pre.substr(1, in_left.length());//左孩子的前序
	string pre_right = pre.substr(in_left.length() + 1);//右孩子的前序

	if (pre_left.length() != 0)//这里不用再看左子树的中序是否为0,因为左子树的中序长度跟左子树的前序长度是相同的
	{
		root->leftchild = new Node(pre_left[0]);
		build(root->leftchild, pre_left, in_left);
	}

	if (pre_right.length() != 0)
	{
		root->rightchild = new Node(pre_right[0]);
		build(root->rightchild, pre_right, in_right);
	}
}

/*
检验是否创建成功
递归遍历前序二叉树
*/
void PreOrderTraverse(Node *&t)
{
	if (t)//结束递归循环的条件
	{
		cout << t->val;
		PreOrderTraverse(t->leftchild);
		PreOrderTraverse(t->rightchild);
	}
}
/*
检验是否创建成功
递归遍历中序二叉树
*/
void InOrderTraverse(Node *&t)
{
	if (t != NULL)
	{
		InOrderTraverse(t->leftchild);
		cout << t->val;
		InOrderTraverse(t->rightchild);
	}
}
/*
检验是否创建成功
递归遍历后序二叉树
*/
void PostOrderTraverse(Node *&t)
{
	if (t != NULL)
	{
		PostOrderTraverse(t->leftchild);
		PostOrderTraverse(t->rightchild);
		cout << t->val;
	}
}

int main()
{
	string pre, in;
	cout << "please input the preorder " << endl;
	cin >> pre;
	cout << "please input the inorder " << endl;
	cin >> in;
	Node *root = new Node(pre[0]);
	build(root, pre, in);
	cout << endl << "输出前序遍历" << endl;
	PreOrderTraverse(root);
	cout << endl << "输出中序遍历" << endl;
	InOrderTraverse(root);
	cout << endl << "输出后序遍历" << endl;
	PostOrderTraverse(root);

	getchar(); getchar();
	return 0;
}