본문 바로가기
알고리즘/트리

[백준][Python] 1991번 트리 순회

by 임짠짠 2022. 2. 25.
반응형
 

1991번: 트리 순회

첫째 줄에는 이진 트리의 노드의 개수 N(1 ≤ N ≤ 26)이 주어진다. 둘째 줄부터 N개의 줄에 걸쳐 각 노드와 그의 왼쪽 자식 노드, 오른쪽 자식 노드가 주어진다. 노드의 이름은 A부터 차례대로 알파

www.acmicpc.net

 

코드

import sys

n = int(input())
tree = {}
for _ in range(n):
    a,b,c = sys.stdin.readline().split()
    tree[a] = [b,c]

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

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

def post(root):
    if root != '.':
        post(tree[root][0])
        post(tree[root][1])
        print(root,end='')
pre('A')
print()
inorder('A')
print()
post('A')
반응형

댓글