forked from doocs/leetcode
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathSolution.py
23 lines (22 loc) · 906 Bytes
/
Solution.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
indexes = {}
def buildTree(self, inorder: List[int], postorder: List[int]) -> TreeNode:
def build(inorder, postorder, i1, i2, p1, p2):
if i1 > i2 or p1 > p2:
return None
root_val = postorder[p2]
pos = self.indexes[root_val]
root = TreeNode(root_val)
root.left = None if pos == i1 else build(inorder, postorder, i1, pos - 1, p1, p1 - i1 + pos - 1)
root.right = None if pos == i2 else build(inorder, postorder, pos + 1, i2, p1 - i1 + pos, p2 - 1)
return root
n = len(inorder)
for i in range(n):
self.indexes[inorder[i]] = i
return build(inorder, postorder, 0, n - 1, 0, n - 1)