알고리즘/프로그래머스

[위클리 챌린지] 5주차

래울 2021. 9. 11. 19:25

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

 

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

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

programmers.co.kr

 

 

- 풀이

 

from itertools import product
def solution(word):
    return sorted(sum([[''.join(j) for j in product(['A','E','I','O','U'], repeat=i+1)] for i in range(5)], [])).index(word)+1