[BOJ] 1991. 트리 순회 (Python)

1991번: 트리 순회

 

난이도: Silver I

문제

---

이진 트리를 입력받아 전위 순회(preorder traversal), 중위 순회(inorder traversal), 후위 순회(postorder traversal)한 결과를 출력하는 프로그램을 작성하시오.

예를 들어 위와 같은 이진 트리가 입력되면,

• 전위 순회한 결과 : ABDCEFG // (루트) (왼쪽 자식) (오른쪽 자식)
• 중위 순회한 결과 : DBAECFG // (왼쪽 자식) (루트) (오른쪽 자식)
• 후위 순회한 결과 : DBEGFCA // (왼쪽 자식) (오른쪽 자식) (루트)

가 된다.

 

입력

---

첫째 줄에는 이진 트리의 노드의 개수 N(1 ≤ N ≤ 26)이 주어진다. 둘째 줄부터 N개의 줄에 걸쳐 각 노드와 그의 왼쪽 자식 노드, 오른쪽 자식 노드가 주어진다. 노드의 이름은 A부터 차례대로 알파벳 대문자로 매겨지며, 항상 A가 루트 노드가 된다. 자식 노드가 없는 경우에는 .으로 표현한다.

 

출력

---

첫째 줄에 전위 순회, 둘째 줄에 중위 순회, 셋째 줄에 후위 순회한 결과를 출력한다. 각 줄에 N개의 알파벳을 공백 없이 출력하면 된다.

 

 

✏️ Solution

---

import sys

input = sys.stdin.readline

def preorder(v):
    if v == '.':
        return 
    print(v, end='')
    preorder(tree[v][0])
    preorder(tree[v][1])

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

def postorder(v):
    if v == '.':
        return
    postorder(tree[v][0])
    postorder(tree[v][1])
    print(v, end='')

n = int(input())

tree = {}
for _ in range(n):
    root, v1, v2 = input().split()
    tree[root] = [v1, v2]

preorder('A')
print()
inorder('A')
print()
postorder('A')

 

전위 순회, 중위순회, 후위순회에 대해 알고 있으면 쉽게 해결할 수 있는 문제였다.

 

나는 각각 순회 방법을 재귀형식으로 구현하여 함수로 만들었다.

 

그리고 입력을 받아 tree를 만들었고 preorder와 inorder 그리고 postorder 함수를 실행하고 결과를 출력하여 정답을 맞을 수 있었다.

---