剑指Offer面试题6 重建二叉树

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

二叉树结点定义如下：

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

题目分析：二叉树的前序遍历顺序是根左右，第一个数为根的值。中序遍历顺序是左根右，根结点的值在序列中间。因此当我们遍历中序遍历找到根结点的值的位置，根结点位置前的数字为左子树结点的值，根结点位置后的数字为右子树结点的值。这样我们就可以找到左右子树的前序遍历和中序遍历序列，接下来可以用同样的方法用递归去构建左右子树。

BinaryTreeNode* ConstructCore(int* startPreorder,int* endPreorder,int* startInorder,int* endInorder);

BinaryTreeNode* Construct(int *preorder,int* inorder,int length)
{
    if(preorder == NULL || inorder == NULL || length <= 0) return NULL;

    return ConstructCore(preorder,inorder,preorder+length-1,inorder+length-1);
}

BinaryTreeNode* ConstructCore(int* startPreorder,int* endPreorder,int* startInorder,int* endInorder)
{
    //前序遍历序列的第一个数字是根结点的值
    int rootValue = startInorder[0];
    BinaryTreeNode* root = new BinaryTreeNode();
    root->m_nValue = rootValue;
    root->m_pLeft = root->m_pRight = NULL;

    if(startPreorder == endPreorder)
    {
        if(startInorder == endInorder && *startPreorder == *startInorder)
            return root;
        else
            cout << "invalid input.";
    }
    //在中序遍历中找到根结点的值
    int* rootInorder = startInorder;
    while(rootInorder  <= endInorder && *rootInorder != rootValue)
        ++rootInorder;
    if(rootInorder == endInorder && *rootInorder != rootValue)
        cout << "invalid input.";
    int leftLength = rootInorder-startInorder;
    int* leftPreorderEnd = startPreorder+leftLength;
    if(leftLength > 0)
    {
        //构建左子树
        root->m_pLeft = ConstructCore(startPreorder+1,leftPreorderEnd,startInorder,rootInorder-1);
    }
    if(leftLength < endPreorder - startPreorder)
    {
        //构建右子树
        root->m_pRight = ConstructCore(leftPreorderEnd+1,endPreorder,rootInorder+1,endInorder);
    }
    return root;
}