https://leetcode.cn/problems/construct-binary-tree-from-inorder-and-postorder-traversal/
给定两个整数数组 inorder 和 postorder ,其中 inorder 是二叉树的中序遍历, postorder 是同一棵树的后序遍历,请你构造并返回这颗 二叉树 。
示例 1:
输入:inorder = [9,3,15,20,7], postorder = [9,15,7,20,3]
输出:[3,9,20,null,null,15,7]
# 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, inorder: List[int], postorder: List[int]) -> Optional[TreeNode]:
# 1. 特殊情况: 树为空,递归终止
if not inorder: # 写中序数组还是后序数组为空,都行
return
# 2. 找到根节点:后序遍历末位就是根节点
rootval = postorder[-1]
root = TreeNode(rootval)
# 3. 切割中序数组:在中序数组中找到根节点的位置,以此为切割点,切割中序数组
index = inorder.index(rootval)
leftinorder = inorder[:index]
rightinorder = inorder[index + 1:]
# 4. 切割后序数组:中序数组左右区间大小一定跟后序数组左右区间大小相同,以len为切割点,切割后序数组
# 思考:为啥切完中序还要切后序,因为要递归用
leftpostorder = postorder[:len(leftinorder)]
rightpostorder = postorder[len(leftinorder):-1]
# 5. 递归
root.left = self.buildTree(leftinorder, leftpostorder)
root.right = self.buildTree(rightinorder, rightpostorder)
# 6. 返回答案
return root