LeetCode 105.前序遍历和中序遍历构造二叉树 C++

前序：中左右

中序：左中右

使用unordermap，可以提高效率

具体解释见注释

/**
 * 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:
    // 定义一个哈希表 用于存储题目所提供的中序遍历内每个节点与其对应的下标
    unordered_map<int, int> pos;
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        int n = preorder.size();
        // 构建哈希表 存储题目所提供的中序遍历内每个节点与其对应的下标 提高速度，否则在中序中找根节点得逐一遍历
        for (int i = 0; i < n; ++i){
            pos[inorder[i]] = i;
        }
        // 参数：先序遍历、中序遍历、先序遍历的起点和终点、中序遍历的起点和终点
        return dfs(preorder, inorder, 0, n-1, 0, n-1);
    }

    TreeNode* dfs(vector<int> preorder, vector<int> inorder, int pl, int pr, int il, int ir){
        if (pl > pr) return NULL;

        
        int val = preorder[pl];// 在前序遍历找到根节点的值
        int k = pos[val];// 在哈希表中找到根节点在中序遍历数组中的位置
        int len = k - il;// k为根节点在中序遍历中的位置，其左边都是根节点的左子树，k-il为左子树长度
        
        auto root = new TreeNode(val);// 建立一个新节点来存储根节点
        root->left = dfs(preorder, inorder, pl+1, pl + len, il, k-1);
        root->right = dfs(preorder, inorder, pl+len+1, pr, k+1, ir);
        return root;
    }
};