Leetcode 889. 根據前序和後序遍歷構造二叉樹

題目連結： https://leetcode-cn.com/contest/weekly-contest-98/problems/construct-binary-tree-from-preorder-and-postorder-traversal/

返回與給定的前序和後序遍歷匹配的任何二叉樹。

 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]

提示：

• 1 <= pre.length == post.length <= 30
• pre[] 和 post[] 都是 1, 2, ..., pre.length 的排列
• 每個輸入保證至少有一個答案。如果有多個答案，可以返回其中一個。

前序遍歷的第一個元素，後續遍歷的最後一個元素，是根節點；

從前序看 2 是左樹的根節點，我們需要知道左樹的長度，我們從後續找到2的位置，4，5，2 是整個左樹。

# Definition for a binary tree node.
class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None


class Solution:
    def constructFromPrePost(self, pre, post):
        """
        :type pre: List[int]
        :type post: List[int]
        :rtype: TreeNode
        """
        tree_root = TreeNode(pre[0])
        pre = pre[1:]
        post = post[:-1]
        len_left = 0
        for i in post:
            if i == pre[0]:
                len_left += 1
                break
            else:
                len_left += 1
        # print(pre, post)
        # print(len_left, pre[:len_left], pre[len_left:])
        if len_left >= 1:
            tree_root.left = self.constructFromPrePost(
                pre[:len_left], post[:len_left])
        if len(pre) - len_left >= 1:
            tree_root.right = self.constructFromPrePost(
                pre[len_left:], post[len_left:])
        return tree_root


print(Solution().constructFromPrePost(
    pre=[1, 2, 4, 5, 3, 6, 7],
    post=[4, 5, 2, 6, 7, 3, 1]))