-
Notifications
You must be signed in to change notification settings - Fork 2
/
145.py
71 lines (56 loc) · 1.9 KB
/
145.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
# [ LeetCode ] 145. Binary Tree Postorder Traversal
def solution(root: "TreeNode") -> list[int]:
def postorder_traverse(node: TreeNode) -> None:
nonlocal answer
if node:
if node.left:
postorder_traverse(node=node.left)
if node.right:
postorder_traverse(node=node.right)
answer.append(node.val)
answer: list[int] = []
postorder_traverse(node=root)
return answer
if __name__ == "__main__":
class TreeNode:
def __init__(
self,
val: int,
left: "TreeNode" = None,
right: "TreeNode" = None
) -> None:
self.val = val
self.left = left
self.right = right
def create_binary_tree(index: int, items: list[int]) -> TreeNode:
if items[index]:
root: TreeNode = TreeNode(val=items[index])
if (index*2+1) < len(items):
root.left: TreeNode | None = create_binary_tree(
index=index*2+1, items=items
)
if (index*2+2) < len(items):
root.right: TreeNode | None = create_binary_tree(
index=index*2+2, items=items
)
return root
else:
return None
cases: list = [
{
"input": { "items": [1, None, 2, None, None, 3, None] },
"output": [3, 2, 1]
},
{
"input": { "items": [None] },
"output": []
},
{
"input": { "items": [1] },
"output": [1]
}
]
for case in cases:
assert case["output"] == solution(
root=create_binary_tree(index=0, items=case["input"]["items"])
)