二叉树专题--AcWing 18. 重建二叉树（利用前、中序遍历，构建二叉树）

思路和代码均类似与上一题 AcWing 1497. 树的遍历，此处就不做过多赘述了。

代码：

class Solution {
    
public:
	#define map unordered_map
    map<int, int> d;
    vector<int> pre, in;


    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
    
        pre = preorder, in = inorder;
        int n = inorder.size();
        for (int i = 0; i < n; ++i)
        {
    
            d[in[i]] = i;
        }
        return dfs(0, n - 1, 0, n - 1);
    }

    TreeNode* dfs(int pl, int pr, int il, int ir)
    {
    
        if (pl > pr) return nullptr;
        auto root = new TreeNode(pre[pl]);
        int k = d[root->val];
        TreeNode* le, * ri;
        le = dfs(pl + 1, k - il + pl, il, k - 1);
        ri = dfs(k - il + pl + 1, pr, k + 1, ir);
        root->left = le, root->right = ri;
        return root;
    }
};