[Cos Pro 1급, Python] 4차 1번 : 사전에서 단어찾기

문제

 

어떤 단어가 XX 사전의 몇 번째 단어인지 알고 싶습니다. XX 사전에는 대문자 알파벳 'A', 'E', 'I', 'O', 'U'를 사용해 만들 수 있는 길이가 5 이하인 모든 단어가 수록되어 있습니다.

예를 들어, 사전의 첫 번째 단어는 "A"이고, 그다음은 "AA"입니다. 마지막 단어는 "UUUUU"입니다.

문자열 word가 매개변수로 주어질 때, word가 사전의 몇 번째 단어인지 return 하도록 solution 함수를 작성했습니다. 그러나, 일부 코드가 잘못되어 코드가 바르게 동작하지 않습니다. 주어진 코드가 모든 입력을 바르게 처리하도록 코드를 수정해주세요. 코드는 한 줄만 수정해야 합니다.

 

 

 

 

 

 

 

 

 

코드

def create_words(lev, s):
	global words
	VOWELS = ['A', 'E', 'I', 'O', 'U']
	words.append(s)
	for i in range(0, 5):
		if lev < 5:
			create_words(lev+1, s + VOWELS[i])

def solution(word):
	global words
	words = []
	answer = 0
	create_words(0, '')
	for idx, i in enumerate(words):
		if word == i:
			answer = idx
			break
	return answer
    
    
    
    
word1 = "AAAAE"
ret1 = solution(word1)

print("solution 함수의 반환 값은", ret1, "입니다.")

word2 = "AAAE"
ret2 = solution(word2)

print("solution 함수의 반환 값은", ret2, "입니다.")