106.从中序与后序遍历序列构造二叉树
根据一棵树的中序遍历与后序遍历构造二叉树。假设树中没有重复的元素。
中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
3
/ \
9 20
/ \
15 7
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def buildTree(self, inorder: List[int], postorder: List[int]) -> TreeNode:
if inorder:
# 用后序遍历找到根结点，index将根节点定位
idx = inorder.index(postorder.pop())
root = TreeNode(inorder[idx])
# 用根结点在中序遍历中切开左右子树，递归重建二叉树
# 需要从右到左构建树，因为我们弹出postorder的最后一个元素。
root.right = self.buildTree(inorder[idx + 1:], postorder)
root.left = self.buildTree(inorder[:idx], postorder)
return root
"""
需要从右到左构建树！！！！！！！！
后序遍历顺序为 左, 右, 根，因此后序遍历的末尾一定为根节点
中序列表为 [左子树中序序列, 根节点, 右子树中序序列]
后序列表为 [左子树后序序列, 右子树后序序列, 根节点]
由于树中没有重复元素，我们可以通过 index 函数确定根节点在中序列表的位置，
进而确定左右子树各自包含的节点总数，将中序与后序列表划分开
每次递归生成根节点并继续递归左右子节点即可
"""
