99클럽 - 모음사전

프로그래머스

아이디어

5개의 모음을 5개의 자리에 중복을 허용하여 놓았을 때
제시되는 단어가 몇 번째로 오는지를 반환하는 문제이다.

그렇다. 중복순열이다.
수학에 젬병이라 수학 문제를 그리 좋아하지는 않는데 내가 풀 수 있다면 이야기가 달라지지. 😁

풀이

def solution(word):
    answer = 0
    data = ["A", "E", "I", "O", "U"]
        
    for idx in range(len(word)):
        answer += (data.index(word[idx])) * (5 ** (5 - idx) - 1) // 4 + 1
    
    return answer

귀찮아서 배열을 사용했는데 더 귀찮았으면 문자열을 사용했을 거고...
데이터의 크기가 커지면 당연히 딕셔너리를 쓰는 것이 빠르니 알아두자.