重建二叉树

题目描述

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

我的代码：

class Solution {
public:
     TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
          if(pre.empty() || vin.empty())
             return NULL;
         TreeNode* pRoot = new TreeNode(pre[0]);
         vector<int> vleft,vright,pleft,pright;
         int len = vin.size();
         int i = 0;
         for(; i < len; i++)
         {
             if(vin[i] == pRoot->val)
                 break;
             vleft.push_back(vin[i]);
             pleft.push_back(pre[i+1]);
        }//i的个数为 该跟的左子树
         for(int j = i+1; j < len; j++)
         {
             vright.push_back(vin[j]);
             pright.push_back(pre[j]);
         }
         pRoot->left = reConstructBinaryTree(pleft,vleft);
         pRoot->right = reConstructBinaryTree(pright,vright);
         return pRoot;
    }
};

思路：前序遍历的第一个节点一定是根节点，在中序遍历中找到这个节点，左边的为左子树，右边的为右子树。

将左子树的前序和中序传入函数返回的值为根的左孩子

将右子树的前序和中序传入参数返回的值为根的右孩子

这样递归的找，就会重建成二叉树。