[프로그래머스/위클리 챌린지] 5주차 (C++)

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

코딩테스트 연습 - 5주차_모음사전

word의 길이가 최대 n이므로 DFS를 이용하여 모든 경우의 수를 구할 수 있다고 판단했다.

3주차는 너무 어렵다...

 

#include <string>
#include <vector>
#include <algorithm>
using namespace std;

string s = "AEIOU";
vector<string> v;

void DFS(int len, string now) {
    if(len == 5) return;

    for(int i = 0; i < s.size(); i++) {
        string tmp = "";
        tmp += s[i];
        v.push_back(now + tmp);
        DFS(len + 1, now + tmp);
    }
} 

int solution(string word) {
    DFS(0, ""); // 현재 만들어진 단어의 길이, 현재 만들어진 단어
    return find(v.begin(), v.end(), word) - v.begin() + 1; // 인덱스 번호 반환
}