[파이썬][백준] 1991번: 트리 순회

1. 문제

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

1991번: 트리 순회

문제 유형: 트리

 

 

2. 소스코드

import sys


class Node:
    def __init__(self, data, left, right):
        self.data = data
        self.left = left
        self.right = right


def preorder(node):
    print(node.data, end='')
    if node.left != '.':
        preorder(tree[node.left])
    if node.right != '.':
        preorder(tree[node.right])


def inorder(node):
    if node.left != '.':
        inorder(tree[node.left])
    print(node.data, end='')
    if node.right != '.':
        inorder(tree[node.right])


def postorder(node):
    if node.left != '.':
        postorder(tree[node.left])
    if node.right != '.':
        postorder((tree[node.right]))
    print(node.data, end='')


n = int(sys.stdin.readline())
tree = {}
for i in range(n):
    data, left, right = map(str, sys.stdin.readline().split())
    tree[data] = Node(data, left, right)

preorder((tree['A']))
print()
inorder(tree['A'])
print()
postorder(tree['A'])
print()