본문 바로가기

알고리즘

99클럽 코테 스터디 16일차 TIL

문제 설명



풀이 과정 & 코드

  • 해당 문제는 word에 주어진 단어가 모음만을 사용해서 알파벳 순서로 몇번쨰에 위치하는지 찾는 문제임
  • itertools에 product를 사용하면 쉽게 풀리는 문제임.
  • 만약 좀 더 좋은 풀이를 생각해본다면 길이가 5인 단어를 끝까지 만들기 전에 word를 찾아서 index를 return 하면 될것같은데 정렬이 되어 있지 않아서 이걸 해결할 방법까지는 크게 떠오르지 않는것 같음...
def solution(word):
    from itertools import product
    answer = []
    vowels = ['a', 'e', 'i', 'o', 'u']
    for i in range(1, 6):
        answer += [''.join(perm) for perm in product(vowels, repeat=i)]
    answer.sort()
    return answer.index(word.lower())+1