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

题目-中等难度

给定两个整数数组 inorder 和 postorder ，其中 inorder 是二叉树的中序遍历， postorder 是同一棵树的后序遍历，请你构造并返回这颗 二叉树 。

示例

示例 1:

输入：inorder = [9,3,15,20,7], postorder = [9,15,7,20,3]
输出：[3,9,20,null,null,15,7]

示例 2:

输入：inorder = [-1], postorder = [-1]
输出：[-1]

提示:

• 1 <= inorder.length <= 3000
• postorder.length == inorder.length
• -3000 <= inorder[i], postorder[i] <= 3000
• inorder 和 postorder 都由 不同 的值组成
• postorder 中每一个值都在 inorder 中
• inorder 保证是树的中序遍历
• postorder 保证是树的后序遍历

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/summary-ranges
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

1. dfs

时间
56ms
击败 89.29%使用 Python3 的用户
内存
20.20MB
击败 85.26%使用 Python3 的用户

# 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]:
        lpost = len(postorder)
        lin = len(inorder)
        dic = defaultdict(int)
        for i in range(len(inorder)):
            dic[inorder[i]] = i
        def helper(postorder: List[int], postLeft: int, postRight: int, d: dict, inLeft: int, inRight: int):
            if postLeft > postRight or inLeft > inRight:
                return None
            rootVal = postorder[postRight]
            root = TreeNode(rootVal)
            pIndex = d[rootVal]
            root.left = helper(postorder,postLeft,pIndex-inLeft+postLeft-1,d,inLeft,pIndex-1)
            root.right = helper(postorder,pIndex-inLeft+postLeft,postRight-1,d,pIndex+1,inRight)
            return root
        return helper(postorder,0,lpost-1,dic,0,lin-1)