본문 바로가기

PS/프로그래머스

[프로그래머스 LV 2] 모음사전 (파이썬/python)

728x90

https://programmers.co.kr/learn/courses/30/lessons/84512

 

코딩테스트 연습 - 모음사전

사전에 알파벳 모음 'A', 'E', 'I', 'O', 'U'만을 사용하여 만들 수 있는, 길이 5 이하의 모든 단어가 수록되어 있습니다. 사전에서 첫 번째 단어는 "A"이고, 그다음은 "AA"이며, 마지막 단어는 "UUUUU"입니

programmers.co.kr

 

 

나는 일단 완전탐색으로 풀었다

 

A AA AAA AAAA AAAAA AAAAE AAAAI ~~~~~를 미리 구해서 리스트에 넣어놓고

 

입력값 word가 몇 번째 인덱스인지 구했다

 

다른분들 보면 5의 제곱근으로 푸셨던데 일단 나는 내가 풀때 그건 생각안나서 그냥 완전탐색했다

 

어짜피 총 경우가 5*6*6*6*6 밖에안되기때문이다.

 

 

 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
def solution(word):
    answer = 0
 
    word_list = []
    words = "AEIOU"
 
    # 모든 단어조합의 경우를 찾는 함수
    def all_word(cnt, w):
        if cnt == 5:
            return
 
        for i in range(len(words)):
            word_list.append(w + words[i])
            all_word(cnt + 1, w + words[i])
 
 
    all_word(0"")
 
 
    return word_list.index(word) + 1
cs

 

 

참고로 all_word함수를 구동한 후 word_list를 print해보면

이렇게 모든 경우가 순서대로 들어간다

 

728x90