[BOJ] 1991 / 트리 순회 / Python 파이썬

[BOJ] 1991 / 트리 순회 / Python 파이썬

[BOJ] 1991 / 트리 순회 / Python 파이썬

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

1991번: 트리 순회

 

풀이

트리 순회의 기본 문제이다.

 

• 전위 순회(pre-order): Root -> Left -> Right
• 중위 순회(in-order): Left -> Root -> Right
• 후위 순회(post-order): Left -> Right -> Root

 

N = int(input())
tree = {}

for _ in range(N):
    node, left, right = input().split()
    tree[node] = [left, right]


def preOrder(node):
    if node == '.':
        return

    print(node, end="")
    preOrder(tree[node][0])
    preOrder(tree[node][1])


def inOrder(node):
    if node == '.':
        return

    inOrder(tree[node][0])
    print(node, end="")
    inOrder(tree[node][1])


def postOrder(node):
    if node == '.':
        return

    postOrder(tree[node][0])
    postOrder(tree[node][1])
    print(node, end="")


preOrder('A')
print()
inOrder('A')
print()
postOrder('A')