剑指 Offer 07. 重建二叉树

'''
输入某二叉树的前序遍历和中序遍历的结果，请构建该二叉树并返回其根节点。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]

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

# 思路： 采用递归法，对于每一颗树首先通过前序遍历的第一个元素值确定根节点，然后利用这个值和中序遍历序列将左子树和右子树区分开。
# 对新的左子树和右子树重复上一过。加一张树的结构图比较容易理解，

 

class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None


class Solution:
    def buildTree(self, preorder, inorder) -> TreeNode:
        def myBuildTree(preorder_left: int, preorder_right: int, inorder_left: int, inorder_right: int):  # 参数为索引下表
            # 构建哈希表，方便在中序遍历中快速查找根节点
            index = {v: k for k, v in enumerate(inorder)}  # 通过字典推导式构建哈希表。将字典的键和值互换，将元素值作为键，索引作为字典的值。这样在index中输入元素值后，输出就是索引

            if preorder_left > preorder_right or inorder_left > inorder_right:  # 递归退出条件
                return None
            # 先构建根节点
            root = TreeNode(preorder[preorder_left])  # 前序遍历的第一个元素是树的根节点
            inorder_root = index[preorder[preorder_left]]  # 得到根节点元素数值在中序遍历中的索引下标
            left_tree_long = inorder_root - inorder_left  # 左子树中的元素个数=中序根节点的索引-中序开头
            # 构造左子树
            root.left = myBuildTree(preorder_left + 1, preorder_left + left_tree_long, inorder_left, inorder_root - 1)
            # （前序遍历的左子树开始，前序遍历的左子树结束，中序遍历的左子树的开始，中序遍历的左子树的结束）
            # 左子树需要去除根节点，所以第一个值要+1（向后取一个元素），左子树最右侧为左子树+1+（树长-1-中序的开始）=preorder_left+left_tree_long
            # 构造右子树
            root.right = myBuildTree(preorder_left + left_tree_long + 1, preorder_right, inorder_root + 1,inorder_right)
            #（前序遍历的右子树开始，前序遍历的右子树结束，中序遍历的右子树的开始，中序遍历的右子树的结束）
            return root

        n = len(preorder)
        return myBuildTree(0, n - 1, 0, n - 1)  # 最开始的索引值从0-n-1


preorder = [3, 9, 20, 15, 7]  # 前序遍历结果
inorder = [9, 3, 15, 20, 7]  # 中序遍历结果
test = Solution()
test.buildTree(preorder,inorder)