重建二叉树

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

• 分析

熟悉二叉树前序&中序遍历规则
前序遍历规则：遍历根节点—->遍历根节点的左子树—->遍历根节点的右子树
中序遍历规则：遍历根节点左子树—->遍历根的节点—->遍历根节点的右子树

前序：1 2 4 3 5 6 中序：4 2 1 5 3 6
从遍历规则可知：前序遍历结果：根节点一定是遍历的第一个结点
中序遍历结果：根节点一定在左子树之后右子树之前遍历
因此，可以从前序遍历结果中找到二叉树的根，从中序遍历结果中找到二叉树的左右子树

• 思路 ：

a. 取前序遍历一个元素，即根节点中数据
b. 在中序遍历结果中找到a中所取元素的位置，按照该位置将序列分为左右两部分，即根的左右子树
c. 创建二叉树的根节点(数据为a中所取数据)
按照b中所划分区间：递归创建根左子树

/**
 * Def inition 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(const vector<int>&pre,size_t &preIndex,const vector<int>&vin,size_t left,size_t right)
    {
        if(preIndex>=pre.size() || left>=right)
            return NULL;

        size_t rootIndex=left;
        // 在中序遍历结果中找二叉树根的位置
        while(rootIndex<right)
        {
            if(pre[preIndex]==vin[rootIndex])
                break;
            rootIndex++;
        }

        // 还原二叉树根节点
        TreeNode *pRoot=new TreeNode(pre[preIndex]);

        // 如果左区间存在，还原根的左子树
        if(left<rootIndex)
            pRoot->left=_reConstructBinaryTree(pre,++preIndex,vin,left,rootIndex);
        // 如果右区间存在，还原根的右子树
        if(rootIndex+1<right)
            pRoot->right=_reConstructBinaryTree(pre,++preIndex,vin,++rootIndex,right);

        return pRoot;
    }

    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        if(pre.size()!=vin.size())
            return NULL;
        size_t preIndex=0;
        return _reConstructBinaryTree(pre,preIndex,vin,0,vin.size());
    }
};