7、重建二叉树

题目描述

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

思路

1、前序遍历序列的第一个元素是根结点root；

2、遍历中序遍历序列，找到根结点root的位置，分割序列，root前面的序列为左子树序列，root后面的序列为右子树序列；

3、根据步骤2中左、右子树的长度，在前序遍历序列中也可以分割出对应左、右子树序列；

4、对左、右子树递归执行上述步骤，直到序列中只有一个元素（即为根结点）。

若已知后序遍历序列与中序遍历序列，思路是一样的。（重构二叉树，必须含有中序序列，才能分割开左、右子树）

C++代码实现

注意：序列分割时，下标表示的关系；

           递归终止条件，只有一个元素（即只有根结点）时，返回。

TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        int m = pre.size();
        int n = vin.size();
        if(m != n || m == 0)
            return NULL;
        return constructCore(pre, vin,0, m-1, 0, n-1);

    }
    TreeNode* constructCore(vector<int> pre, vector<int> vin, int l1, int r1, int l2, int r2)//参数为前序序列首尾，中序序列首尾
    {
        TreeNode* root = new TreeNode(pre[l1]);//先序第一个结点为根结点
        if(l1 == r1)//只有一个结点
            return root;
        //在中序序列中找根结点
        int index;
        for(index = l2; index <= r2; index++)
        {
            if(vin[index] == root->val)
                break;
        }
        //分割中序序列，求出左子树、右子树的序列长度
        int length_left = index - l2;
        int length_right = r2 - index;
        if(length_left > 0)
            //构建左子树
            root ->left = constructCore(pre ,vin, l1+1, l1+length_left, l2, index - 1);
        if(length_right > 0)
            //构建右子树
            root -> right = constructCore(pre, vin, l1+length_left+1, r1, index+1, r2);
        return root;
    }