Return any binary tree that matches the given preorder and postorder traversals.
Values in the traversals pre
and post
are distinct positive integers.
Example 1:
Input: pre = [1,2,4,5,3,6,7], post = [4,5,2,6,7,3,1] Output: [1,2,3,4,5,6,7]
Note:
1 <= pre.length == post.length <= 30
pre[]
andpost[]
are both permutations of1, 2, ..., pre.length
.- It is guaranteed an answer exists. If there exists multiple answers, you can return any of them.
# 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:
n = len(preorder)
if n == 0:
return None
root = TreeNode(preorder[0])
if n == 1:
return root
for i in range(n - 1):
if postorder[i] == preorder[1]:
root.left = self.constructFromPrePost(
preorder[1: 1 + i + 1], postorder[: i + 1])
root.right = self.constructFromPrePost(
preorder[1 + i + 1:], postorder[i + 1: -1])
return root