[ programmers ] 위클리 챌린지 5주차 84512번 ( python )

문제

프로그래머스 84512번 위클리 챌린지 5주차 풀이 ( 파이썬 )

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

코딩테스트 연습 - 5주차

 

풀이

단어들을 중복 순열로 만들어 리스트를 만들어 정렬한 후 리스트에서 해당 word를 찾았다.

from itertools import product

def solution(word):
    answer = 0

    words = ['A', 'E', 'I', 'O', 'U']

    dict_list = []

    for i in range(1, 6):
        dict_list += list(map("".join, product(words, repeat = i)))
    
    dict_list.sort()

    for i in range(len(dict_list)):
        if dict_list[i] == word:
            answer = i + 1
    
    return answer

풀고 나니까 생각났는데 해당 word를 찾을 때 이분탐색을 이용했으면 더 빠를거라고 생각이 들었다.