BOJ 1991 : 트리 순회 [C++]

문제

1991번: 트리 순회

 

 

'인덱스를 알파벳으로 해야겠구나'라고 생각해버려서 map을 썼다.

근데 다른 사람들 풀이를 보니 이차원배열 혹은 pair의 배열을 만들고 인덱스를

(대문자 알파벳 - 'A')로 써서 해결했던데 이걸 왜 생각을 못했나 싶다.

 

#include <iostream>
#include <map>

std::map<char, std::pair<char, char>> edges;

std::string pre_str = "";
std::string in_str = "";
std::string post_str = "";

void traversal(char node) {
    pre_str += node;
    if (edges[node].first != '.') {
        traversal(edges[node].first);
    }
    in_str += node;
    if (edges[node].second != '.') {
        traversal(edges[node].second);
    }
    post_str += node;
}

int main() {
    int n = 0;
    std::cin >> n;

    for (int i = 0; i < n; i++) {
        char parent, left, right;
        std::cin >> parent >> left >> right;
        edges.insert({parent, {left, right}});
    }

    traversal('A');

    std::cout << pre_str << "\n" << in_str << "\n" << post_str << "\n";

    return 0;
}