[Swift] lv.1 콜라 문제 | Programmers

문제 설명

오래전 유행했던 콜라 문제가 있습니다. 콜라 문제의 지문은 다음과 같습니다.

정답은 아무에게도 말하지 마세요.

콜라 빈 병 2개를 가져다주면 콜라 1병을 주는 마트가 있다. 빈 병 20개를 가져다주면 몇 병을 받을 수 있는가?

단, 보유 중인 빈 병이 2개 미만이면, 콜라를 받을 수 없다.

문제를 풀던 상빈이는 콜라 문제의 완벽한 해답을 찾았습니다. 상빈이가 푼 방법은 아래 그림과 같습니다. 우선 콜라 빈 병 20병을 가져가서 10병을 받습니다. 받은 10병을 모두 마신 뒤, 가져가서 5병을 받습니다. 5병 중 4병을 모두 마신 뒤 가져가서 2병을 받고, 또 2병을 모두 마신 뒤 가져가서 1병을 받습니다. 받은 1병과 5병을 받았을 때 남은 1병을 모두 마신 뒤 가져가면 1병을 또 받을 수 있습니다. 이 경우 상빈이는 총 10 + 5 + 2 + 1 + 1 = 19병의 콜라를 받을 수 있습니다.

문제를 열심히 풀던 상빈이는 일반화된 콜라 문제를 생각했습니다. 이 문제는 빈 병 a개를 가져다주면 콜라 b병을 주는 마트가 있을 때, 빈 병 n개를 가져다주면 몇 병을 받을 수 있는지 계산하는 문제입니다. 기존 콜라 문제와 마찬가지로, 보유 중인 빈 병이 a개 미만이면, 추가적으로 빈 병을 받을 순 없습니다. 상빈이는 열심히 고심했지만, 일반화된 콜라 문제의 답을 찾을 수 없었습니다. 상빈이를 도와, 일반화된 콜라 문제를 해결하는 프로그램을 만들어 주세요.

콜라를 받기 위해 마트에 주어야 하는 병 수 a, 빈 병 a개를 가져다 주면 마트가 주는 콜라 병 수 b, 상빈이가 가지고 있는 빈 병의 개수 n이 매개변수로 주어집니다. 상빈이가 받을 수 있는 콜라의 병 수를 return 하도록 solution 함수를 작성해주세요.

제한 사항

1 ≤ b < a ≤ n ≤ 1,000,000
정답은 항상 int 범위를 넘지 않게 주어집니다.

1차 코드

import Foundation

func solution(_ a: Int, _ b: Int, _ n: Int) -> Int {
    var now = n
    var count = 0
    var received = 0

    while now >= a {
        received = now / a
        count += received * b
        now = received + (now % a)
    }
    return count
}

1차 결과

음 그렇다. 어느 특정 조건만 만족하고 나머지는 모두 나가리가 되었다.

while에서 빠그러지는 것 말고는 문제될 것이 없으니 직접 while 루프를 따라가면서 계산해봐야겠다.

받은 병의 갯수를 바탕으로 while문 내의 now를 조정했어야했는데 그냥 received 갯수만 추가하니까 틀렸나보다. 3개를 갖다주면 2개를 받는 경우도 있을텐데 빼고 했었다.

2차 코드

import Foundation

func solution(_ a: Int, _ b: Int, _ n: Int) -> Int {
    var now = n
    var count = 0
    var received = 0

    while now >= a {
        received = now / a
        count += received * b
        now = (now % a) + (received * b)
    }
    return count
}

2차 결과

보완점

received는 while문 내에서 변하지 않는 값인데 굳이 밖에다가 var로 선언할 이유가 없었다.

let으로 내부에서 선언하면 해당 범위에서만 작동하는 변수 혹은 상수임을 알 수 있어서 더 좋을 것 같다고 생각이 들어 위치를 옮겼다.

최종 코드

import Foundation

func solution(_ a: Int, _ b: Int, _ n: Int) -> Int {
    var now = n
    var count = 0

    while now >= a {
        let received = now / a
        count += received * b
        now = (now % a) + (received * b)
    }
    return count
}