题目
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中不含有重复地数字。例如输入的前序遍历序列{1 , 2 , 4 , 7 , 3, 5 , 6 , 8}和中序遍历序列{4 , 7 , 2 , 1 , 5 , 3 , 8 , 6},则重建出下图1所示的二叉树并输出它的头结点。
(1)
/ \
(2) (3)
/ / \
(4) (5) (6)
\ /
(7) (8)
图1
分析
在二叉树的前序遍序列中,第一个数字总是树的根结点的值。但在中序遍历序列中,根结点的值在序列的中间,左子树的结点的值位于根结点的值的左边,而右子树的结点的值位于根结点的值的右边。因此需要扫描中序遍历序列,才能找到根结点的值。
如上图,前序遍历序列的第一个数字1就是根结点的值。扫面中序遍历序列,就能确定根结点的值的位置。根据中序遍历序列特点,在根结点的值1前面的数字都是左子树结点的值,位于1后面的数字都是右子树结点的值。
由于中序遍历序列中,有3个数字是左子树结点的值,因此左子树总共有3个有3个做子结点。同样,在前序遍历序列中,根结点后面的3个数字就是3个左子树的结点的值,在后面的所有数字都是右子树结点的值。这样就在前序和中序这两个遍历中,分别找到左右子树的对应的子序列。
代码
#include<stdio.h>
#include<iostream>
#include <exception>
using namespace std;
typedef int ElemType;
typedef struct BiTNode {
ElemType data; //结点的数据域
struct BiTNode *lchild, *rchild; //左右孩子
}BiTNode, *BiTree;
/*递归的调用constructCore去构造左右子数*/
BiTree constructCore(int* sPreorder, int* ePreorder, int* sInorder, int* eInorder){
//前序遍历序列中的第一个数字是根结点
int rootValue = sPreorder[0];
BiTNode* root = new BiTNode();
root->data = rootValue;
root->lchild = NULL;
root->rchild = NULL;
//判断是否只有一个根结点
if(sPreorder == ePreorder) {
if(sInorder == eInorder && *sPreorder == *sInorder)
return root;
else
throw std::exception("Invaild input");
}
//在中序遍历寻找根结点的值
int* rootInorder = sInorder;
while(rootInorder <= eInorder && *rootInorder != rootValue)
rootInorder++;
if(rootInorder == eInorder && *rootInorder != rootValue)
throw std::exception("Invalid input");
//左子树的数量
int leftLength = rootInorder - sInorder;
int* leftPreorderEnd = sPreorder + leftLength;
if(leftLength > 0) {
//构建左子树
root->lchild = constructCore(sPreorder+1,leftPreorderEnd,sInorder,rootInorder - 1);
}
if(leftLength < ePreorder - sPreorder) {
//构建右子树
root->rchild = constructCore(leftPreorderEnd+1,ePreorder,rootInorder+1,eInorder);
}
return root;
}
BiTree construct(int* preorder, int* inorder , int length) {
if(preorder == NULL || inorder == NULL || length <= 0)
return NULL;
return constructCore(preorder, preorder+length-1,inorder,inorder+length-1);
}
/*后序遍历递归函数*/
void postOrderTraverse(BiTree T) {
if(T) {
postOrderTraverse(T->lchild);
postOrderTraverse(T->rchild);
cout<<T->data;
}
}
void main () {
int preorder[8] = {1, 2, 4, 7, 3, 5, 6, 8};
int inorder[8] = {4, 7, 2, 1, 5, 3, 8, 6};
BiTNode* root = new BiTNode;
root = construct(preorder, inorder, 8);
printf("后序遍历测试\n");
postOrderTraverse(root);
cout<<"\n\n";
}