剑指 Offer 07. 重建二叉树-根据前序、中序遍历构建二叉树，返回根节点

一、题目

二、思路

三、代码

/**
 * 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) {
        //将中序遍历添加到哈希表中方便查找
        unordered_map<int,int>dic;
        for(int i=0;i<inorder.size();++i)
        {
            dic.emplace(inorder[i],i);
        }
        //传入参数到dfs,容器，根节点，左边界，右边界
        return dfs(preorder,inorder,dic,0,0,inorder.size()-1);
    }
    TreeNode* dfs(vector<int>& preorder, vector<int>& inorder,unordered_map<int,int>&dic,int root,int left,int right)
    {
        if(left>right)
        {
            return nullptr;
        }
        TreeNode *head=new TreeNode(preorder[root]);
        int len=dic[preorder[root]];
        //左子树：
        //根节点：root+1
        //左边界：left
        //右边界：len-1
        head->left=dfs(preorder,inorder,dic,root+1,left,len-1);
        //右子树：
        //根节点：root+len-left+1
        //左边界：len+1
        //右边界：right
        head->right=dfs(preorder,inorder,dic,root+len-left+1,len+1,right);
        return head;
    }
};