백준 1991(트리 순회) - python

접근

입력이 문자로 들어오니 dict 을 사용해서 트리를 구성해야겠다.

 

 

나의 코드

import sys
input = sys.stdin.readline

T = {}

n = int(input())

for _ in range(n):
    s,*string = list(map(str,input().split()))
    T[s] = string

def pre(start):
    if start != '.':
        print(start,end='')
        pre(T[start][0])
        pre(T[start][1])

def inorder(start):
    if start != '.':
        inorder(T[start][0])
        print(start,end='')
        inorder(T[start][1])

def po(start):
    if start != '.':
        po(T[start][0])
        po(T[start][1])
        print(start,end='')

pre('A')
print()
inorder('A')
print()
po('A')