백준 1991번 트리 순회

정보

문제 바로가기 [클릭]

난이도: Silver1

관련 개념: #트리 #순회

---

조건

시간 제한	메모리 제한
2 초	128 MB

---

문제

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

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

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

---

입력

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

---

출력

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

---

예제 입출력 1

입력

7
A B C
B D .
C E F
E . .
F . G
D . .
G . .

출력

ABDCEFG
DBAECFG
DBEGFCA

---

코드(파이썬)

def preorder(parent):
    if all(ch == '.' for ch in tree[parent]):
        return parent
    else:
        result = ''
        l, r = tree[parent]
        
        result += parent
        if l != '.':
            result += preorder(l)
        if r != '.':
            result += preorder(r)
            
        return result
    
def inorder(parent):
    if all(ch == '.' for ch in tree[parent]):
        return parent
    else:
        result = ''
        l, r = tree[parent]
        
        if l != '.':
            result += inorder(l)
        result += parent
        if r != '.':
            result += inorder(r)
            
        return result
    
def postorder(parent):
    if all(ch == '.' for ch in tree[parent]):
        return parent
    else:
        result = ''
        l, r = tree[parent]
        
        if l != '.':
            result += postorder(l)
        if r != '.':
            result += postorder(r)
        result += parent
            
        return result
        
n = int(input())
tree = dict()

for p, l, r in [input().split() for _ in range(n)]:
    tree[p] = (l, r)
    
print(preorder('A'))
print(inorder('A'))
print(postorder('A'))

---

특이사항

• 간단한 트리 순회 문제

---

참고문헌

-