https://leetcode.cn/problems/construct-binary-tree-from-preorder-and-inorder-traversal/
给定两个整数数组 preorder 和 inorder ,其中 preorder 是二叉树的先序遍历, inorder 是同一棵树的中序遍历,请构造二叉树并返回其根节点。
示例 1:
输入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
输出: [3,9,20,null,null,15,7]
class Solution:
def buildTree(self, preorder: List[int], inorder: List[int]) -> Optional[TreeNode]:
# ⼆叉树的构造问题⼀般都是使⽤「分解问题」的思路:构造整棵树 = 根节点 + 构造左⼦树 + 构造右⼦树。
# 可以通过前序或者后序遍历结果找到根节点,然后根据中序遍历结果确定左右⼦树。
# 1、讨论特殊情况
if not inorder: # 写中序数组还是前序数组为空,都行
return
# 2、找到根节点,前序遍历时根节点就是首位
rootval = preorder[0]
root = TreeNode(rootval)
# 3、切割中序数组,在中序数组中找到根节点的位置,以此为切割点,切割中序数组
index = inorder.index(rootval)
leftinorder = inorder[:index]
rightinorder = inorder[index + 1:]
# 4、切割前序:中序数组左右区间大小一定跟前序数组左右区间大小相同,以len为切割点,切割前序数组
# 思考:为啥切完中序还要切前序,因为要递归用
leftpreorder = preorder[1:len(leftinorder) + 1]
rightpreorder = preorder[len(leftinorder) + 1:]
# 5、递归
root.left = self.buildTree(leftpreorder, leftinorder)
root.right = self.buildTree(rightpreorder, rightinorder)
# 6、返回
return root