剑指 Offer 07. 重建二叉树

题目描述

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

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

示例 1:

Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]
示例 2:

Input: preorder = [-1], inorder = [-1]
Output: [-1]

限制：

0 <= 节点个数 <= 5000

分析

在二叉树的前序遍历序列中，第一个数字总是树的根节点的值。但是在中序遍历序列中，根节点的值在序列的中间，左子树的节点的值位于根节点的值的左边，右子树的节点的值位于根节点的值的右边。

所以我们要先开一个hash表来记录中序遍历的下标。

然后在前序遍历中找到根节点，根据hash表找到中序遍历中根节点的位置，这样就能确定左右子树的范围。然后递归构建左右子树。

代码

/**
 * 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> hash;
    TreeNode* Build(vector<int>& preorder, vector<int>& inorder, int pre_l, int pre_r, int in_l, int in_r) {
        if (pre_l > pre_r) return NULL;
        int h_root = hash[preorder[pre_l]];
        int left_size = h_root - in_l;
        TreeNode* root = new TreeNode(inorder[h_root]);
        root->left = Build(preorder, inorder, pre_l + 1, pre_l + left_size, in_l, h_root - 1);
        root->right = Build(preorder, inorder, pre_l + 1 + left_size, pre_r, h_root + 1, in_r);
        return root;
    }

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