剑指offer-4 -- 重建二叉树 - C++

题目描述：

输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

例如，给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树：

    3
   / \
  9  20
    /  \
   15   7

限制：0 <= 节点个数 <= 5000

根节点preorder[0]在中序遍历数列中的位置index，index左边的部分为左子树的中序遍历序列，index右边的为右子树的中序遍历序列。相应的在前序遍历中，从索引1开始后的部分为对应的左子树和右子树的前序遍历序列，它们的元素个数和中序遍历中划分后的个数相同。 然后将左子树和右子树分别当作一棵新的二叉树，递归调用函数进行重建。

class Solution {
public:
    TreeNode* rebuild(vector<int>& pre, int pre_left, int pre_right, vector<int>& vin, int vin_left, int vin_right){
        if(pre_left > pre_right)
            return nullptr;
        TreeNode* root = new TreeNode(pre[pre_left]);
        int root_index = vin_left;
        while(root_index <= vin_right && vin[root_index] != pre[pre_left])
            root_index += 1;
        root->left = rebuild(pre, pre_left+1, pre_left+root_index-vin_left, vin, vin_left, root_index-1);
        root->right = rebuild(pre, pre_left+root_index-vin_left+1, pre_right, vin, root_index+1, vin_right);
        return root;
    }
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        return rebuild(pre, 0, pre.size()-1, vin, 0, vin.size()-1);
    }
};