[백준 5032] 탄산 음료 (Swift)

5032번: 탄산 음료

예제를 보고 “9 0 3 인데 어떻게 3이 아니라 4가 답이지??” 라며 생각하다가, 구글링을 해서야 무슨 문제인지 이해했다.

빈 병의 수가 9개이고, 새 병 1개로 바꾸는 데 필요한 빈 병의 수가 3개일 때

• 9 / 3 = 3개의 새 병을 받을 수 있음
• 다 마시면 3개의 빈 병이 됨
• 따라서 3 / 3 = 1개의 새 병을 받아 또 마실 수 있음.
• 그래서 총 4개를 마실 수 있는 것!

즉 새 병으로 바꿀 때 마다 그 몫을 빈 병 수에 더해줘야 한다. (다 마시면 빈병이 되므로)

풀이1 (틀림)

let inputArr = readLine()!.split(separator: " ").map { Int($0)! }
var bottle = inputArr[0] + inputArr[1]
let exchange = inputArr[2]
var cola = 0
while bottle >= exchange {
    cola += bottle / exchange // 몫
        cola += bottle % exchange // 나머지
    bottle = bottle / exchange
}
print(cola)

처음에 손으로 나눗셈해보니 결과가 ‘매 나눗셈의 몫과 나머지를 모두 더한 것’ 과 같길래 이렇게 풀었었다.

하지만 바로 틀렸고..

예제 1,2는 모두 통과했기 때문에 이 풀이가 맞는 줄 알았는데, 몫을 빈 병에 더해주지 않았기 때문에 계산이 틀렸다.

반례로, 5 0 3 을 넣어 보면 정답(2)이 나오지 않아 틀린 풀이이다.

풀이2

let inputArr = readLine()!.split(separator: " ").map { Int($0)! }
var bottle = inputArr[0] + inputArr[1]
let exchange = inputArr[2]
var cola = 0
while bottle >= exchange {
    cola += bottle / exchange // 몫
    bottle = bottle / exchange + bottle % exchange // 바꾼 콜라도 다 먹고 빈병에 다시 더해야 함
}
print(cola)

병의 수에 몫과 나머지를 모두 더해주어야 다음 시행 때 정확한 빈 병 갯수로 계산할 수 있다.