[백준-1991] 트리 순회 / Python

1991번: 트리 순회

import sys

input = sys.stdin.readline
n = int(input())
tree = {}
for _ in range(n):
    root, left, right = map(str, input().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")

 

해설

재귀 함수로 해결할 수 있는 문제.

전위, 중위, 후위 순회 별로 출력 위치를 다르게 해주면 된다.