-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path트리 순회.py
39 lines (30 loc) · 976 Bytes
/
트리 순회.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
import sys
n = int(sys.stdin.readline())
tree = {}
for n in range(n):
root, left, right = sys.stdin.readline().split()
tree[root] = [left, right]
# 전위 순회
def preorder(root):
if root != '.':
print(root, end='') # 루트 노드 출력
preorder(tree[root][0]) # 왼쪽 서브트리 순회
preorder(tree[root][1]) # 오른쪽 서브트리 순회
# 중위 순회
def inorder(root):
if root != '.':
inorder(tree[root][0]) # 왼쪽 서브트리 순회
print(root, end='') # 루트 노드 출력
inorder(tree[root][1]) # 오른쪽 서브트리 순회
# 후위 순회
def postorder(root):
if root != '.':
postorder(tree[root][0]) # 왼쪽 서브트리 순회
postorder(tree[root][1]) # 오른쪽 서브트리 순회
print(root, end='') # 루트 노드 출력
preorder('A')
print()
inorder('A')
print()
postorder('A')
print()