프로그래머스 - 모음사전(Swift)

프로그래머스 - 모음사전(Swift)

 

 

 문제 설명

 

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

프로그래머스

 

 나의 풀이

 

5개의 원소만 돌면 되기 때문에 완전탐색 DFS로 문제를 풀었다.

 

import Foundation

func solution(_ word:String) -> Int {
    var dic: [String: Int] = [:]
    
    func DFS(_ L: Int, _ accStr: String) {
        if L != 0 {
            cnt += 1
            dic[accStr] = cnt    
        }
        if L == arr.count {
            return
        } else {
            for i in 0..<5 {
                DFS(L+1, accStr + arr[i])
            }
        }
    }
    
    let arr: [String] = ["A", "E", "I", "O", "U"]
    var cnt = 0
    DFS(0, "")
    
    
    return dic[word]!
}