[프로그래머스/C++] 모음사전

미숙한 블로그 주인이 코딩테스트 문제를 풀어가는 과정을 담은 글입니다. 이 풀이가 효율적인 풀이가 아닐 수 있으며, 부정확한 정보가 많이 있을 수 있습니다. 보완해야할 점이 있다면 댓글로 남겨주세요!

 

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

코딩테스트 연습 - 모음사전

모음사전

문제

사전에 알파벳 모음 'A', 'E', 'I', 'O', 'U'만을 사용하여 만들 수 있는, 길이 5 이하의 모든 단어가 수록되어 있습니다. 사전에서 첫 번째 단어는 "A"이고, 그다음은 "AA"이며, 마지막 단어는 "UUUUU"입니다.

단어 하나 word가 매개변수로 주어질 때, 이 단어가 사전에서 몇 번째 단어인지 return 하도록 solution 함수를 완성해주세요.

제한사항

• word의 길이는 1 이상 5 이하입니다.
• word는 알파벳 대문자 'A', 'E', 'I', 'O', 'U'로만 이루어져 있습니다.

 

풀이

입출력 예시

 

풀이

그냥 모든 경우의 수를 저장해서 몇 번째인지 찾을까 생각을 잠깐 했지만 아무래도 이건 좀 아니다 싶었다.

 

규칙을 찾기 위해 보다 작은 곳에서 부터 시작해보자.

 

1. 'A', 'E' 만을 이용해서 만든 길이 2 이하의 단어
A AA AE
E EA EE

전체 경우의 수 : 6
AA -> AE : +1 ( 의미 : 두 번째 단어가 1번 바뀌면 + 1번째 단어가 됨. )
A -> E : +3 = 2 + 1 ( 의미 : 첫 번째 단어가 1번 바뀌면 + 3번째 단어가 됨. )

EA 는 몇 번째?
첫 번째 A가 E로 변함. 1 + 3
두 번째 A가 그대로임. 1
4 + 1 = 5
5번째.

2. 'A', 'E', 'I' 만을 이용해서 만든 길이 3 이하의 단어
A AA AAA AAE AAI AE AEA AEE AEI AI AIA AIE AII
E EA ... EII
I IA ... III

전체 경우의 수 : 39
AAA -> AAE : +1
AA -> AE : +4 = 3 + 1
A -> E : +13 = 4 * 3 + 1 = 3 * ( 3 + 1 ) + 1

IAE는 몇 번째? IAI IE IEA IEE IEI II IIA IIE III
첫 번째 A가 I로 변함. ( 1 + 13 + 13 )
두 번째 A가 그대로임. 1
세 번째 A가 E로 변함. 1 + 1
27 + 1 + 2 = 30
30번째.

규칙이 보이기 시작한다. 마지막으로 한 번만 더 해보자.

3. 'A', 'E', 'I', 'O' 만을 이용해서 만든 길이 4 이하의 단어
A AA AAA AAAA AAAE AAAI AAAO AAE ... AOOO
E ... EOOO
I ... IOOO
O ... OOOO

전체 경우의 수 : 340
AAAA -> AAAE : +1 = 1
AAA -> AAE : +5 = 4 + 1
AA -> AE : +21 = 4 * ( 4 + 1 ) + 1
A -> E : +85 = 4 * ( 4 * ( 4 + 1 ) + 1 ) + 1

IOEE 는 몇 번째?
A -> I = 1 + 85 + 85
A -> O = 1 + 21 + 21 + 21
A -> E = 1 + 5
A -> E = 1 + 1
171 + 64 + 6 + 2 = 243
243번째

그럼 'A', 'E', 'I', 'O', 'U' 만을 이용해서 만든 길이 5 이하의 단어도 같은 규칙으로 알 수 있다.

 

#include <string>
#include <vector>
#include <map>

using namespace std;

int solution(string word) {
    int answer = 0;
    map<char, int> alp;
    int next_word[5] = { 781, 156, 31, 6, 1 };
    
    alp.insert({'A', 0});
    alp.insert({'E', 1});
    alp.insert({'I', 2});
    alp.insert({'O', 3});
    alp.insert({'U', 4});
    
    int i = 0;
    for (const auto& w : word) {
        answer += 1 + alp[w] * next_word[i];
        i++;
    }
    
    return answer;
}