牛客66题（4）重建二叉树

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

/**
 * 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 vinlength=vin.size();
        vector<int> left_pre,right_pre,left_in,right_in;
        if(vinlength==0)
        {
            return NULL;
        }
         TreeNode* node=new TreeNode(pre[0]);//建立一个新的节点为根节点。
        
        int gen=0;
        for(int i=0;i<vinlength;i++)
        {
            if(vin[i]==pre[0])
            { gen=i;
                break;}
        }//记录根节点位置
        for(int i=0;i<gen;i++)
        {
            left_in.push_back(vin[i]);//将根节点左侧中序序列压入容器
                left_pre.push_back(pre[i+1]);//将除了根节点的先序遍历压入容器
        }
        for(int i=gen+1;i<vinlength;i++)
        {
            right_in.push_back(vin[i]);//根节点右侧中序压入容器
            right_pre.push_back(pre[i]);//根节点右侧先序压入容器
        }//至此有了四个容器分别存放根节点左右先序和中序遍历的节点信息
//利用递归，调用的函数每次返回根节点。
        node->left=reConstructBinaryTree(left_pre,left_in);
        node->right=reConstructBinaryTree(right_pre,right_in);
        return node;
    }
};

总结：题目稍难，首先搞懂先序遍历与中序遍历的联系，先序遍历的首元素是根节点，在中序遍历中找到根节点位置，左侧为左子树，右侧为右子树。先将根结点找到后建立二叉树的节点，将根结点左侧右侧的先序中序都压到容器里面，递归调用自身，形成二叉树。