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

📃 문제

[위클리 챌린지] 모음사전 🔗링크

사전에 알파벳 모음 'A', 'E', 'I', 'O', 'U'만을 사용하여 만들 수 있는, 길이 5 이하의 모든 단어가 수록되어 있습니다. 사전에서 첫 번째 단어는 "A"이고, 그다음은 "AA"이며, 마지막 단어는 "UUUUU"입니다.
단어 하나 word가 매개변수로 주어질 때, 이 단어가 사전에서 몇 번째 단어인지 return 하도록 solution 함수를 완성해주세요.

• 조건
  1. 주어지는 단어. 1 ≤ word의 길이 ≤ 5
  2. word를 이루는 요소 = 'A','E','I','O','U'

🧠 풀이

✏️ word보다 사전순이 빠른 단어를 셀 수 있다면, word가 몇번째 단어인지 알 수 있다.
✏️ 주어진 word는 A,E,I,O,U 5개의 단어로 이루어져 있고 중복해서 단어를 사용할 수 있다.
따라서 중복 순열을 이용하면 만들 수 있는 단어의 수를 구할 수 있다.

• 1자릿수 이하 단어 : 5파이1
• 2자릿수 이하 단어 : 5파이1 + 5파이2
• 3자릿수 이하 단어 : 5파이1 + 5파이2 + 5파이3
• 4자릿수 이하 단어 : 5파이1 + ... + 5파이4
• 5자릿수 이하 단어 : 5파이1 + ... + 5파이5

E 이전에 나열되는 단어는 A로 시작하는 5자릿수 이하 단어이고, 첫번째 자리가 A로 고정되기 때문에
E 이전에 나열되는 단어의 수 = A로 시작하는 5자릿수 이하 단어 = 4자릿수 이하 단어
E의 순서 = E 이전에 나열되는 단어의 수 + 1

✏️ 예시

• word = 'IE'
  → IE의 순서 = IE 이전에 나열되는 단어의 수 + 1
  IE의 순서 = A로 시작하는 5자릿수 이하 단어 + E로 시작하는 5자릿수 이하 단어 + IA로 시작하는 5자릿수 이하 단어 + 1
  IE의 순서 = 4자릿수 이하 단어 + 4자릿수 이하 단어 + 3자릿수 이하 단어 + 1
  IE의 순서 = (4자릿수 이하 단어 * 2) + 3자릿수 이하 단어 + 1

• word = 'EAE'
  → EAE의 순서 = EAE 이전에 나열되는 단어의 수 + 1
  EAE의 순서 = A로 시작하는 5자릿수 이하 단어 + EAA로 시작하는 5자릿수 이하 단어 + 1
  EAE의 순서 = 4자릿수 이하 단어 + 2자릿수 이하 단어 + 1

def solution(word):
    word_num = {'A':0,'E':1,'I':2,'O':3,'U':4}
    step = [0]
    
    for digit in range(0,5):
        multi_permutation = 5 ** digit
        step.insert(0, step[0] + multi_permutation)
    step.pop()
    
    answer = 0 
    for idx,w in enumerate(word):
        answer += word_num[w] * step[idx] + 1
        
    return answer