leetcode_106_从中序与后序遍历序列构造二叉树

从中序与后序遍历序列构造二叉树

描述

中等

根据一棵树的中序遍历与后序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

示例

中序遍历 inorder = [9,3,15,20,7]

后序遍历 postorder = [9,15,7,20,3]

返回如下的二叉树：

    3
   / \
  9  20
    /  \
   15   7

解题

和第105题类似

节点分布

• 后序遍历 [左子树节点 右子树节点 根节点]
• 中序遍历 [左子树节点 根节点 右子树节点]

然后也是递归建树

# 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, inorder: List[int], postorder: List[int]) -> TreeNode:
        if not postorder:
            return None
        # 后序遍历中最后一个节点为根节点
        x = postorder[-1]
        root = TreeNode(x)
        # 找到中序遍历中该节点的位置
        index = inorder.index(x)
		# 递归建树
        root.left = self.buildTree(inorder[:index], postorder[:index])
        root.right = self.buildTree(inorder[index+1:], postorder[index:-1])

        return root