52. Leetcode 889. 根据前序和后序遍历构造二叉树 (二叉树-二叉树构建)

返回与给定的前序和后序遍历匹配的任何二叉树。

 pre 和 post 遍历中的值是不同的正整数。

 

示例：

输入：pre = [1,2,4,5,3,6,7], post = [4,5,2,6,7,3,1]
输出：[1,2,3,4,5,6,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 constructFromPrePost(self, preorder: List[int], postorder: List[int]) -> TreeNode:
        def dfs(pre, post):
            if not pre:
                return None

            # 数组长度为1时，直接返回结果
            if len(pre) == 1:
                return TreeNode(pre[0])

            # 根据前序数组的第一个元素，创建根节点
            root = TreeNode(pre[0])

            # 根据前序数组第二个元素，确定后序数组左子树范围
            left_count = post.index(pre[1]) + 1

            # 递归执行前序数组左边、后序数组左边
            root.left = dfs(pre[1:left_count+1], post[:left_count])
            root.right = dfs(pre[left_count+1:], post[left_count:-1])

            return root

        return dfs(preorder, postorder)