JZ7 重建二叉树

给定节点数为 n 的二叉树的前序遍历和中序遍历结果，请重建出该二叉树并返回它的头结点。

例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建出如下图所示。

提示:

1.vin.length == pre.length

2.pre 和 vin 均无重复元素

3.vin出现的元素均出现在 pre里

4.只需要返回根结点，系统会自动输出整颗树做答案对比

数据范围：n≤2000n \le 2000n≤2000，节点的值 −10000≤val≤10000-10000 \le val \le 10000−10000≤val≤10000

要求：空间复杂度 O(n)O(n)O(n)，时间复杂度 O(n)O(n)O(n)

解析：

/**

 * Definition for binary tree

 * struct TreeNode {

 *     int val;

 *     TreeNode *left;

 *     TreeNode *right;

 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}

 * };

 */

class Solution {

public:

    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {

        int n = pre.size();

        int m = vin.size();

        if(n==0 || m==0)

        {

            return nullptr;

        }

        //创建根

        TreeNode* root = new TreeNode(pre[0]);

        //遍历输入的中序数组

        for(int i=0;i<vin.size();i++)

        {

            //找到根节点

            if(pre[0] == vin[i])

            {

                //左子树的前序遍历

                vector<int> leftpre(pre.begin()+1,pre.begin()+i+1);

                //左子树的中序遍历

                vector<int> leftvin(vin.begin(),vin.begin()+i);

                //构建左子树

                root->left = reConstructBinaryTree(leftpre, leftvin);

                //右子树的前序遍历

                vector<int> rightpre(pre.begin()+i+1,pre.end());

                //右子树的中序遍历

                vector<int> rightvin(vin.begin()+i+1,vin.end());

                //构建右子树

                root->right = reConstructBinaryTree(rightpre, rightvin);

                //为空就结束

                break;

            }

        }

        return root;

    }

};