剑指offer-JZ4 重建二叉树

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

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    unordered_map<int, int> pos;//用图来存储中序遍历中各个节点的位置，用空间来换时间，这样子，不用每次都去遍历查找
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
    	//把中序遍历中的节点值作为键，在中序遍历中的下标号作为值
        int n=pre.size();
        for(int i=0;i<n;i++)
        {
            pos[vin[i]]=i;
        }
        //参数的含义依次是：前序遍历数组，中序遍历数组，当前根节点前序遍历的左端点，前序遍历的右端点，中序遍历的左端点，中序遍历的右端点
        return dfs(pre,vin,0,n-1,0,n-1);
    }
    TreeNode* dfs(vector<int> pre,vector<int> vin,int pl,int pr,int vl,int vr)
    {
        if(pl>pr) return NULL;//递归截至条件
        TreeNode* root=new TreeNode(pre[pl]);//重建二叉树，每次都要新声明一个二叉树节点。根节点为前序遍历的左端点
        int k=pos[root->val]-vl;//根节点在中序遍历的位置-中序遍历左端点=左子树的大小
        root->left=dfs(pre,vin,pl+1,pl+k,vl,vl+k-1);//以左子树节点为根节点的dfs
        root->right=dfs(pre,vin,pl+k+1,pr,vl+k+1,vr);//以右子树节点为根节点的dfs
        return root;
    }
};