剑指offer--python--重建二叉树

第五题：重建二叉树

solution_from_香辣鸡排蛋包饭

大致思路：递归

前序遍历：

[ 根节点, [左子树的前序遍历结果], [右子树的前序遍历结果] ]

中序遍历：

[ [左子树的中序遍历结果], 根节点, [右子树的中序遍历结果] ]

时间复杂度：O(n)，空间复杂度：O(n)

# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param preOrder int整型一维数组 
# @param vinOrder int整型一维数组 
# @return TreeNode类
#
class Solution:
    def reConstructBinaryTree(self , preOrder: List[int], vinOrder: List[int]) -> TreeNode:
        # write code here
        index = {val:i for i, val in enumerate(vinOrder)} #中序遍历的值val和下标i分别作为key和value存入字典
        def build(preOrderId, vinL, vinR): #前序遍历中根节点下标，该根节点所在的树在中序遍历中左边界，中序遍历右边界
            if vinL > vinR:
                return None
            rootVal = preOrder[preOrderId] #根节点的值是由前序遍历中前序遍历的下标获得
            vinRootId = index[rootVal] #中序遍历中根节点的下标是由字典获得
            leftSize = vinRootId - vinL #左子树的大小是中序遍历中根节点的下标减去左边界
            root = TreeNode(rootVal) 
            root.left = build(preOrderId + 1, vinL, vinRootId - 1) 
            root.right = build(preOrderId + 1 + leftSize, vinRootId + 1, vinR)
            return root
        return build(0, 0, len(vinOrder) - 1)