面试题6：重建二叉树

题目：输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不包含重复的数字。例如输入前序遍历序列{1， 2， 4， 7， 3， 5， 6， 8}和中序遍历{4， 7， 2， 1， 5， 3， 8， 6}，则重建出图2.6所示的二叉树并输出它的头结点。二叉树节点的定义如下：

struct BinaryTreeNode
{
    int value;
    BinaryTreeNode *pLeft;
    BinaryTreeNode *pRight;
};

在二叉树的前序遍历序列中，第一个数字总是树的根节点的值。但在中序遍历序列中，根节点的值在序列的中间，左子树的节点的值位于根节点的值的左边，而右子树的节点的值位于根节点的值的右边。因此我们需要扫描中序遍历序列，才能找到根节点的值。
如图2.7所示，前序遍历序列的第一个数字1就是根节点的值。扫描中序遍历序列，就能确定根节点的值的位置。根据中序遍历特点，在根节点的值1前面的3个数字都是左子树节点的值，位于1后面的数字都是右子树节点的值。

由于在中序遍历中，有3个数字是左子树节点的值，因此左子树总共有3个左子节点。同样，在前序遍历的序列中，根节点后面的3个数字就是3个左子树节点的值，再后面的所有数字都是右子树节点的值。这样我们就在前序遍历和中序遍历两个序列中，分别找到了左右子树对应的子序列。

既然我们已经分辨找到了左、右子树的前序遍历序列和中序遍历序列，我们可以用同样的方法分别取构建左右子树。也就是说，接下来的事情可以用递归的方法去完成。

BinaryTreeNode * ConstructCore(int * startPreorder, int * endPreorder, int * startInorder, int * endInorder)
{
    BinaryTreeNode * root = new BinaryTreeNode();
    root->value = startPreorder[0];
    root->pLeft = root->pRight = NULL;

    if(startPreorder == endPreorder)
    {
        if(startInorder == endInorder && *startPreorder == *startInorder)
            return root;
        else
            throw std::exception("Invalid input.");
    }

    int *rootInorder = startInorder;
    while (rootInorder <= endInorder && *rootInorder != startPreorder[0])
    {
        ++ rootInorder;
    }

    if(rootInorder == endInorder && *rootInorder != startPreorder[0])
        throw std::exception("Invalide input.");

    int leftLength = rootInorder - startInorder;
    int *leftPreorderEnd = startPreorder + leftLength;
    if(leftLength > 0)
        root->pLeft = ConstructCore(startPreorder+1, leftPreorderEnd, startInorder, rootInorder-1);
    if(leftLength < endPreorder - startPreorder)
        root->pRight = ConstructCore(leftPreorderEnd+1, endPreorder, rootInorder+1, endInorder);
    return root;
}

BinaryTreeNode * Construct(int * preorder, int * inorder, int length)
{
    if(preorder == NULL || NULL == inorder || length <= 0)
        return;
    return ConstructCore(preorder, preorder + length - 1, inorder, inorder + length - 1);
}