🟩 [백준] [Python] [Silver1] 1991번_트리 순회

문제

https://www.acmicpc.net/problem/1991

1991번: 트리 순회

풀이

# 백준 1991번 - 트리 순회
n = int(input())
tree = {}
for _ in range(n):
    root, left, right = input().split()
    tree[root] = [left, right]

def preorder(root):
    if root != '.':
        print(root, end='')
        preorder(tree[root][0]) # left
        preorder(tree[root][1]) # right

def inorder(root):
    if root != '.':
        inorder(tree[root][0]) # left
        print(root, end='')
        inorder(tree[root][1]) # right

def postorder(root):
    if root != '.':
        postorder(tree[root][0]) # left
        postorder(tree[root][1]) # right
        print(root, end='')

preorder('A')
print()
inorder('A')
print()
postorder('A')

재귀함수를 통해서 트리를 구현하였다.