Create construct_tree_inorder_preorder.py

This commit is contained in:
bt3gl 2023-07-29 16:13:26 -07:00 committed by GitHub
parent cf4a797646
commit bfeb0bf97c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -0,0 +1,21 @@
def build_tree(preorder: list[int], inorder: list[int]) -> Optional[TreeNode]:
def helper(i_left, i_right, index_map):
if i_left > i_right:
return None
root = TreeNode(preorder.pop(0))
index_here = index_map[root.val]
# this order change from postorder
root.left = helper(i_left, index_here - 1, index_map)
root.right = helper(index_here + 1, i_right, index_map)
return root
index_map = {value: index for index, value in enumerate(inorder)}
return helper(0, len(inorder) - 1, index_map)