[C++]백준 - 1991번 문제

1991번: 트리 순회 (acmicpc.net)

1991번: 트리 순회

 

1991번 : 트리 순회

---

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

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

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

 

입력

---

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

 

 

출력

---

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

 

 

 

---

 

생각해 볼 점

---

전위 순회, 중위 순회, 후위 순위를 차례로 구현합니다.

 

순회는 함수로 구현하였으며, 재귀의 형식으로 진행됩니다.

 

코드

---

#include <iostream>
using namespace std;
pair<int, int> *bin_tree;

//전위 순회
void pre_order(int current)
{    
    char node = current + 65;
    printf("%c", node);
    int left = bin_tree[current].first;
    int right = bin_tree[current].second;
    if(0 <= left) pre_order(left);
    if(0 <= right) pre_order(right);
}

//중위 순회
void in_order(int current)
{
    char node = current + 65;
    int left = bin_tree[current].first;
    int right = bin_tree[current].second;
    if(0 <= left) in_order(left);
    printf("%c", node);
    if(0 <= right) in_order(right);
}

//후위 순회
void post_order(int current)
{
    char node = current + 65;
    int left = bin_tree[current].first;
    int right = bin_tree[current].second;
    if(0 <= left) post_order(left);
    if(0 <= right) post_order(right);
    printf("%c", node);
}



int main() 
{
    int N;
    cin >> N;
    
    //트리
    bin_tree = new pair<int, int>[N];
    
    //대문자 알파벳 - 65 = int
    for(int i = 0; i < N; i++)
    {
        char node, left, right;
        cin >> node >> left >> right;
        int i_node = node - 65;
        bin_tree[i_node] = make_pair(left - 65, right - 65);
    }
    pre_order(0);
    printf("\n");
    in_order(0);
    printf("\n");
    post_order(0);
    
    
    delete[] bin_tree;
	return 0;
}

 

그 외

---