6. 重建二叉树

题目描述：

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

---

解析

根据先序遍历和中序遍历还原二叉树的主要思想：
1、先序遍历序列的第一个元素必定是根节点，可以由此获取二叉树的根节点。
2、根据根节点，在中序遍历序列中查找该节点，由中序遍历的性质可知，中序遍历中该根节点左边的序列必定在根节点的左子树中，而根节点右边的序列必定在右子树中。由此可以知道先序遍历中左子树以及右子树的起止位置。
3、分别对左子树和右子树重复上述的过程，直至所有的子树的起止位置相等时，说明已经到达叶子节点，遍历完毕。

主要代码实现

//========================重建二叉树========================
bTreeNode* 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);
}


//======================重建二叉树核函数==========================
bTreeNode* ConstructCore(int* startPreorder, int* endPreorder, int* startInorder, int* endInorder){
    //前序遍历的第一个节点是根节点，先创建根节点
    int rootValut = startPreorder[0];
    bTreeNode* root = (bTreeNode*)malloc(sizeof(bTreeNode)); //为节点分配存储空间
    root->info = rootValut;
    root->left = NULL;
    root->right = NULL;

    //判断该根节点是否还有左右子树
    if(startPreorder == endPreorder){                                     //前序遍历时候，只有一个根节点
        if(startInorder == endInorder && *startPreorder == *startInorder) //中序遍历时候，也只有一个节点，并且这个节点等于前序遍历时候的根节点
            return root;
        else
            exit(1);      //输入的前序和中序不是同一个二叉树的遍历
    }

    //在中序遍历中找根节点的位置
    int* rootInorder = startInorder;
    while(rootInorder <= endInorder && *rootInorder != rootValut){
        rootInorder++;
    }
    if(rootInorder == endInorder && *rootInorder != rootValut){ //对上面循环退出条件进行判断， 在中序遍历中找不到相应的根节点
        exit(2);  
    }

    //根据根节点在中序遍历中的位置，确定左右子树的数量
    int leftLength = rootInorder - startInorder;
    int* leftPreorderEnd = startPreorder + leftLength;

    //根据确定的左右子树的数量，递归建立左右子树
    if(leftLength > 0)  {
        //构建左子树
        root->left = ConstructCore(startPreorder+1, leftPreorderEnd, startInorder, rootInorder-1);
    }
    if(leftLength < endPreorder - startPreorder){ //判断右子树是否为空
        root->right = ConstructCore(leftPreorderEnd+1, endPreorder, rootInorder+1, endInorder);
    }
    return root;
}