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

문제 링크: https://school.programmers.co.kr/learn/courses/30/lessons/84512

프로그래머스

코드

from itertools import product
def solution(word):
    words = ['A', 'E', 'I', 'O', 'U']
    arr = []
    for i in range(1,6):
        tmp=list(product(words,repeat=i))
        for j in tmp:
            arr.append(''.join(j))
    arr.sort()
    return arr.index(word)+1

풀이

• 문자열이 최대 5자리이므로 중복 순열로 모든 단어의 수를 리스트에 넣고 정렬해도 시간 초과에 걸리지 않는다.
• 모든 문자가 5자리로 이루어진게 아니므로, product의 repeat을 1부터 5자리까지 반복해서 문자를 구해야한다.