[프로그래머스 LV2] - 모음사전(트리) c++

문제

https://school.programmers.co.kr/learn/courses/30/lessons/84512

프로그래머스

접근방법

count를 전역변수로 설정하고,

방문횟수마다 +1을 해서

tree에 저장했다.

 

이렇게 하면,

각 노드의 자식 노드 개수를 구할 수 있다.

코드

#include <string>
#include <vector>
#include <iostream>
#include <map>
using namespace std;

vector<char> dict = {'A', 'E', 'I', 'O', 'U'};
map<string, int> tree;

int count = 1;

void make_tree(string word) {
    for (int i=0;i<5;i++) {
        string temp = word + dict[i];
        if (temp.size() >= 6) return;
        tree[temp] = count;
        count++;
        make_tree(temp);
    }
    return;
}

int solution(string word) {
    int answer = 1;
    make_tree("");
    answer = tree[word];
    return answer;
}

개선할 점