剑指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);
}