[프로그래머스] 콜라문제 - Swift

코드

func solution(_ a:Int, _ b:Int, _ n:Int) -> Int {

    var answer = 0
    
    // 더 받는 콜라 갯수를 answer에 저장하고 현재 콜라 갯수를 다시 함수에 넘겨
    // 재귀를 사용하여 해결
    func numCola (_ z:Int) -> Int {
        // 더 받는 콜라 갯수
        answer += b * (z/a)
        // 지금 콜라 갯수
        var cola = b * (z/a) + z % a
        if cola < a {
            return answer
        } else {
            return numCola(cola)
        }
    }
    return numCola(n)
}

회고

• 수학적 문제였던 것 같다
• 코드 구현은 그렇게 어렵지 않았다
• 재귀 함수 공부를 했었는데 실제로 내 손으로 구현이 되어서 좋았다