[[EVD]] - 剑指 Offer 07. 重建二叉树

题目分析:[[EVD]] - 剑指 Offer 07. 重建二叉树https://leetcode-cn.com/problems/zhong-jian-er-cha-shu-lcof/

简单描述:

• 输入某二叉树的前序遍历和中序遍历的结果，请构建该二叉树并返回其根节点。
• 假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

限制🚫

• 0 <= 节点个数 <= 5000
• **不含重复的数字**

示例:

  3
 / \
9  20
  /  \
15    7
Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]

解题思路:

思路:

• 前序和中序遍历唯一确定一颗二叉树,先序确定根,中序根据根划分左右子树
  • 前序:3(根) | ,9 |,20,15,7
  • 中序:9,|3(根)|,15,20,7
• #分治 #哈希表 本题不含重复的数字
  • 通过先序建树,则存储map[先序root] = 中序root下标
  • (先序数组中) 起始左子树节点 = root + 1,
  • (先序数组中) 起始右子树节点 = root + 1 + 左子树总数(在中序数组中计算)

效率:

• 时间复杂度
• 空间复杂度

代码:

class Solution
{
public:
    /*先序建树*/
    TreeNode *buildTree(vector<int> &preorder/*先序*/, vector<int> &inorder/*中序*/)
    {
        for (int i = 0; i < inorder.size(); i++)
            map[inorder[i]] = i;
        return dg(preorder, 0, 0, inorder.size() - 1);
    }

private:
    unordered_map<int, int> map; //<先序数组中root数值,中序数组下标>
    /*在中序数组中划分左右子树,根据先序的root值确定*/
    TreeNode *dg(vector<int> &preorder, int root /*先序数组root下标*/, int left /*中序数组中*/, int right /*中序数组中*/)
    {
        if (left > right)
            return nullptr;
        TreeNode *cur = new TreeNode(preorder[root]);
        int idx = map[preorder[root]];  // 获得对应在中序数组中的下标
        cur->left = dg(preorder, root + 1, left, idx - 1);  // [left,idx-1]为在中序数组中划分的左子树
        cur->right = dg(preorder, idx - left + root + 1, idx + 1, right); // [idx+1, right]为在中序数组中划分的右子树, idx-left为左子树节点数量
        return cur;
    }
};