[백준 1991] 트리 순회 (python)

1991번: 트리 순회

---

 

Dictionary에 root를 key로 하고, [left, right]를 value로 저장한다.

 

import sys
from collections import defaultdict
input = sys.stdin.readline
result = ""


def preorder(x):
    print(x, end="")
    if arr[x][0] != ".":
        preorder(arr[x][0])
    if arr[x][1] != ".":
        preorder(arr[x][1])


def inorder(x):
    if arr[x][0] != ".":
        inorder(arr[x][0])
    print(x, end="")
    if arr[x][1] != ".":
        inorder(arr[x][1])


def postorder(x):
    if arr[x][0] != ".":
        postorder(arr[x][0])
    if arr[x][1] != ".":
        postorder(arr[x][1])
    print(x, end="")


arr = defaultdict(list)
N = int(input())
for _ in range(N):
    x, y, z = input().split()
    arr[x].extend([y, z])

preorder("A")
print()
inorder("A")
print()
postorder("A")