重建二叉树

题目
输入某二叉树的前序遍历和中序遍历的结果，请构建该二叉树并返回其根节点。

假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

思路
重点是根据根节点，确定左子树和右子树的长度。
根据前序遍历可以确定根节点，根据中序遍历可以确定左右子树的长度，从而确定左子树和右子树的先序遍历和中序遍历序列。
程序可看作二叉树的后序遍历。
代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {

        if(preorder.size()==0)
        return NULL;
        int len=preorder.size();
        TreeNode* root=new TreeNode(preorder[0]);
        if(len==1)
        return root;

        int l_len=0;
        int pivot=preorder[0];
        while(inorder[l_len]!=pivot) l_len++;
        int r_len=len-1-l_len;

        vector<int> l_preorder(preorder.begin()+1,preorder.begin()+1+l_len);
        vector<int> l_inorder(inorder.begin(),inorder.begin()+l_len);
        vector<int> r_preorder(preorder.begin()+1+l_len,preorder.end());
        vector<int> r_inorder(inorder.begin()+l_len+1,inorder.end());
        root->left=buildTree(l_preorder,l_inorder);
        root->right=buildTree(r_preorder,r_inorder);

        return root;
    }
};