重建二叉树

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

思路:

先序遍历第一个位置肯定是根节点node，中序遍历的根节点位置在中间p，在p左边的肯定是node的左子树的中序数组，p右边的肯定是node的右子树的中序数组另一方面，先序遍历的第二个位置到p，也是node左子树的先序子数组，剩下p右边的就是node的右子树的先序子数组把四个数组找出来，分左右递归调用即可.

代码:

/**
 * 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 in_size = vin.size();
 
        if(in_size == 0)  //当中序遍历数组为空时,表示无相应子树,返回子树为空
           return NULL; 
        
        int p = pre[0];  
         TreeNode * node = new TreeNode(p);  //利用先序遍历数组第一个值作为二叉树的头结点
        vector<int>pre_left,pre_right,vin_left,vin_right;
        
        int j=0;
        for(int i = 0;i< vin.size();i++){
            if(vin[i]==p){
             j=i;                    //记录头结点的值在中序遍历数组中的索引
            break;                   //剪枝操作
            }
        }
        
        for(int k=0;k<vin.size();k++){   //把中序遍历数组中头结点左边的数存到vin_left
            if(k<j){
                vin_left.push_back(vin[k]);
            }
            else if(k>j)
            {
             vin_right.push_back(vin[k]); //把中序遍历数组中头结点右边的数存到vin_right
            }
        }
        
        for(int q=1;q<pre.size();q++){
            if(q<=j)
                pre_left.push_back(pre[q]);   //把先序遍历数组中头结点的左子树的值存到 
                                              //pre_left
            else
                pre_right.push_back(pre[q]); //把先序遍历数组中头结点的右子树的值存到 
                                             //pre_right
            
        }
        
       
        
        node->left = reConstructBinaryTree(pre_left,vin_left);//重建头结点的左子树,并返回
                                                            //该左子树的的头结点作为父亲结点 
                                                             //的左孩子

        node->right =reConstructBinaryTree(pre_right,vin_right);//重建头结点的右子树,并返回
                                                            //该右子树的的头结点作为父亲结点 
                                                             //的右孩子
        return node; //返回父亲结点
 
        
        
    }
};