剑指offer---重建二叉树（6）

题目：输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1，2，4，7，3，5，6，8}和中序遍历序列{4，7，2，1，5，3，8，6}；

二叉树的定义如下：

strcut BinaryTreeNode
{
int  m_nValue;
BinaryTreeNode*   m_pLeft;
BinaryTreeNode*   m_pRight;
};

利用递归建立左右二叉树。

具体思想在此不做详细介绍。详情请看相关资料。

#include <iostream>
using namespace std;

struct BinaryTreeNode
{
int  m_nValue;
BinaryTreeNode*   m_pLeft;
BinaryTreeNode*   m_pRight;
};
BinaryTreeNode* ConstructCore(int* startPreorder,int* endPreorder,int* startInorder,int* endInorder )
{
//前序遍历的第一个结点是根结点的值
int rootValue = startPreorder[0];
BinaryTreeNode* root = new BinaryTreeNode();
root->m_nValue = rootValue;
root->m_pLeft = NULL;
root->m_pRight = NULL;
if(startPreorder == endPreorder)
{
if(startInorder == endInorder && *startPreorder == *endPreorder)
{
return root;
}
else
{
throw exception("Invalid input");
}
}
//在中序遍历中找到根结点的值
int* rootInorder = startInorder;
while(rootInorder <= endInorder && *rootInorder != rootValue)
rootInorder++;
if(rootInorder == endInorder && *rootInorder != rootValue)
{
throw exception("Invalid input");
}

int leftlengt = rootInorder-startInorder;
int* leftPreorder = startPreorder + leftlengt;
if(leftlengt > 0)
{
root->m_pLeft = ConstructCore(startPreorder +1,leftPreorder,startInorder,rootInorder-1);
}
if(leftlengt < endPreorder-startPreorder)
{
root->m_pRight = ConstructCore(leftPreorder+1,endPreorder,rootInorder+1,endInorder);
}
return root;
}
BinaryTreeNode* Construct(int* preorder,int* inorder,int length )
{
if(preorder==NULL || inorder==NULL ||length<=0)
{
return NULL;
}
return ConstructCore(preorder,preorder+length-1,inorder,inorder+1);
}