[Python] Programmers. 위클리 챌린지 5주차 - 모음 사전

작성일 | In |

위클리 챌린지 5주차 - 모음 사전

문제 링크

풀이 코드

  • 중복 순열 문제입니다. 파이썬의 itertools.product로 쉽게 풀 수 있습니다.
  • 정렬을 해주는 것을 잊으면 안됩니다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
# 위클리 챌린지 5주차 - 모음 사전
from itertools import product

def solution(word):
    answer = 0
    vowels = ['A','E','I','O','U']
    words = []
    for i in range(1,6):
        words.extend(list(product(vowels, repeat=i))) # 모든 중복순열을 담아줍니다.
        
    words.sort()
    
    for i in range(len(words)):
        words[i] = ''.join(words[i]) # 관리할 수 있는 문자열 형태로 바꿔줍니다.
        
    answer = words.index(word) + 1 # A가 첫번째 단어입니다.
    return answer

비고