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

문제 유형

 한줄 수정

난이도

 normal

Note 

 1. recursive 함수, create_words 로직 암기.

 

Code

# -*- coding: utf-8 -*-
# UTF-8 encoding when using korean
# 4차 1번

words = []

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, "입니다.")

 

※ 가끔 코드 중 print(~)가 있습니다. 정리 못한 점 죄송합니다.