剑指 Offer 07. 重建二叉树-Python题解

输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

例如，给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树：

限制：

0 <= 节点个数 <= 5000

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
        preLen = len(preorder)
        inLen = len(inorder)

        if( preLen !=  inLen):
            raise RuntimeError("parameter  error")
        
        # 将中序遍历的列表转化成字段，k是列表中的值，v是值对应的索引
        inMap = {}
        for i in range(len(inorder)):
            inMap[inorder[i]] = i
        # 构建整棵树
        return self.myBuildTree(preorder, 0, preLen - 1, inMap, 0, inLen - 1)

    # preorder:前序遍历的列表
    # preLeft:前序遍历中左边界
    # preRight:前序遍历中右边界
    # inorder:中序遍历的字典，k是中序遍历的值，v是对应的值在中序遍历中的索引
    # inLeft:中序遍历的左边界
    # inRight:中序遍历的右边界
    def myBuildTree(self, preorder, preLeft, preRight, inorder,  inLeft, inRight):
        # 递归终止条件，如果一棵树的前序遍历的左边界大于右边界，或中序遍历的左边界大于有边界都是不构成区间的
        if( preLeft  > preRight or  inLeft > inRight):
            return  None
        # 前序遍历中左边界对应的值就是根节点的值
        pIndexVal = preorder[preLeft]
        # 通过根节点的值区中序遍历的字典中获取相应的索引
        pIndex = inorder[pIndexVal]
        # 创建一棵树
        root = TreeNode(pIndexVal)
        # 通过递归的方式 来得到左子树
        root.left = self.myBuildTree(preorder, preLeft + 1, pIndex - inLeft + preLeft,
        inorder, inLeft,  pIndex - 1)
        # 通过递归的方式得到右子树
        root.right = self.myBuildTree(preorder, pIndex - inLeft + preLeft + 1, preRight,
        inorder,  pIndex + 1, inRight)
        return root