剑指offer之重建二叉树

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

知识点：

前序遍历：父节点--左子节点--右子节点 （又叫先序遍历，先根遍历）；

中序遍历：左子节点--父节点--右子节点 （又叫中根遍历）；

解决思路：递归，根据前序遍历的第一个数字创建根节点；在中序遍历序列中找到根节点的位置，确定左、右子树节点数量；然后递归构建左、右子树。

具体代码：（C++）

typedef struct TreeNode{

    int val;

    TreeNode* left;

    TreeNode* right;

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

};

class Solution

{

public:

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

    {

        if(pre.empty() || vin.empty() || pre.size() != vin.size())

                return nullptr;

         vector<int> left_pre,right_pre,left_vin,right_vin;

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

         int left_length = 0;

        while(pre[0] != vin[left_length] && left_length < pre.size())

                ++left_length;      //左子树的长度

        for(int i=0;i<left_length;i++)

        {

                left_pre.push_back(pre[i+1]);       //左子树的前序序列 ，pre[0]为根节点

                left_vin.push_back(vin[i]);            //左子树的中序序列

        }

        for(int i=left_length + 1; i<pre.size();++i)

        {

                right_pre.push_back(pre[i]);        //右子树的前序序列

                right_vin.push_back(vin[i]);        //右子树的中序序列

         }

         node->left = reConstructBinaryTree(left_pre,left_vin); //递归创建左子树

          node->right = reConstructBinaryTree(right_pre,right_vin); //递归创建右子树

        return node;

    }

};