[ 프로그래머스 ] 모음사전

 

 

🍄 프로그래머스 문제 링크

프로그래머스

✍🏻 문제 풀이 Github 링크

-Swift-CodingTest/프로그래머스/2/84512. 모음 사전 at main · BaeJihae/-Swift-CodingTest

 

 

문제 설명

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

단어 하나 word가 매개변수로 주어질 때, 이 단어가 사전에서 몇 번째 단어인지 return 하도록 solution 함수를 완성해주세요.

 

 

제한사항

• word의 길이는 1 이상 5 이하입니다.
• word는 알파벳 대문자 'A', 'E', 'I', 'O', 'U'로만 이루어져 있습니다.

 

입출력 예

word	result
"AAAAE"	6
"AAAE"	10
"I"	1563
"EIO"	1189

 

 

문제 풀이

나의 풀이

import Foundation

func solution(_ word:String) -> Int {
    
    let alphabet = ["A","E","I","O","U"]
    var count = 0
    var isAnswer: Bool = false
    
    func dfs(str: String, num: Int) {
        
        if str != word {
            count += 1
        }else {
            isAnswer = true
        }
        
        for w in alphabet {
            if num < 5 && !isAnswer {
                dfs(str: str+w, num: num+1)
            }
        }
    }
    dfs(str: "", num: 0)
    return count
}

 

문제 유형이 완전탐색이라고 해서.. DFS로 푸는 걸 알았다. 완전탐색이라고 적혀있지 않았다면 DFS문제인지 바로 떠오르진 않았을 것 같다. 더듬더듬 “피로도” 문제를 풀었던 기억을 더듬어 dfs로 풀기 시작했다.

 

사전과 동일한 값이 나와서 더이상 반복하지 않아도 될 때의 조건을 적용하는 것이 어려웠다. 간단하게 true/false를 넣을 수 있는 변수하나만 추가하니 word와 동일한 값이 나왔을 때 더이상 재귀함수를 돌지 않도록 설정할 수 있었다.

 

처음엔 str값을 [String] 배열로 주었는데, 그럼 join하는 시간도 있고, append도 해줘야하는데 상수에다가 append할 수가 없어서 이를 String으로 바꾸어 풀어주었다.