백준 1991 트리 순회

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

1991번: 트리 순회

이문제는 트리구조를 받아서 어떤 형태로 저장시켜서 어떻게 이용할껀지에 대해 요구하는 문제

 

흔히 알고있는 dfs, bfs 트리구조로 데이터를 2차원 리스트로 받아 처리하려고 하니까 어려움이 있어

Tuple로 처리를 했다

 

즉,  A라는 루트는 left, right의 자식 노트가 있을 것이고 key, value의 값으로 묶는다는 개념으로 접근하면 쉽다

 

import sys
input =  sys.stdin.readline

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])
        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')