105. 从前序与中序遍历序列构造二叉树

105. 从前序与中序遍历序列构造二叉树

原始题目链接：https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/

给定两个整数数组 preorder 和 inorder ，其中 preorder 是二叉树的先序遍历， inorder 是同一棵树的中序遍历，请构造二叉树并返回其根节点。

解题思路：

构造树的过程使用递归方法，需要考虑递归的结束条件和自身调用。构造的过程和分析先序和中序的过程是一样的，先序遍历的第一个值一定是根节点，然后根据这个节点去中序遍历中查找这个根节点，这样就把一棵树分开成了左子树
根 右子树，然后进行构造，再递归的构造左子树和右子树，具体实现看代码及注释。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
        # 定义一个函数，递归的调用
        # 参数是树的前序遍历的左右边界和中序遍历的左右边界
        def helper(pre_s, pre_e, in_s, in_e):
            # 没有节点了
            if pre_s > pre_e:
                return None
            
            # 前序遍历的第一个节点是根节点
            root_node = preorder[pre_s]
            # 在中序遍历中查找根节点
            root_index = find_in_root_dict[root_node]
            # 建立根节点
            root = TreeNode(root_node)
            # 计算左子树的节点个数：中序遍历中根节点的位置索引 - 中序遍历的最左边的位置索引
            left_tree_size = root_index - in_s

            # 递归构造左子树：传的参数是左子树的先序遍历的左右边界，左子树的中序遍历的左右边界
            # 左子树的先序遍历的左右边界分别是：先序遍历中根节点索引 + 1， 根的索引+左子树的个数
            # 左子树的中序遍历的左右边界分别是：中序遍历的左边界， 中序遍历中根节点的位置-1
            root.left = helper(pre_s + 1, pre_s + left_tree_size, in_s, root_index - 1)

            # 递归构造右子树：传的参数是右子树的先序遍历的左右边界，右子树的中序遍历的左右边界
            # 右子树的先序遍历的左右边界分别是：根的索引 + 左子树的个数 + 1，先序遍历的最右边界
            # 右子树的中序遍历的左右边界分别是：中序遍历中根节点的索引 + 1， 中序遍历的最右边界
            root.right = helper(pre_s + left_tree_size + 1, pre_e, root_index + 1, in_e)

            return root

        # 访问一次所有的中序遍历结果，存到哈希表中，这样在查找根节点的时候可以优化速度
        find_in_root_dict = {node: i for i, node in enumerate(inorder)}    

        # 先序遍历和中序遍历的元素个数都是一样的，算一个就好了
        n = len(preorder)
        return helper(0, n - 1, 0, n - 1)

参考文献：
https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/solution/cong-qian-xu-yu-zhong-xu-bian-li-xu-lie-gou-zao-9/