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

thumbnail

안녕하세요. 이번에는 프로그래머스 콜라 문제를 풀어보려고 합니다.

 

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

프로그래머스

---

Problem

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

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

콜라 빈 병 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. 입출력 예

---

Solution

class Solution {
    fun solution(a: Int, b: Int, n: Int): Int {
        var answer: Int = 0
        var rest: Int = 0
        var copyN = n
        
        while(copyN >= a) {
            if(copyN % a != 0) {
                rest += (copyN % a)
            }
            
            copyN = copyN / a * b
            answer += copyN
            copyN += rest
            rest = 0
        }
        
        return answer
    }
}

 이번 문제는 a개를 가져다주면 b개를 주는 식과 n개에서 콜라로 바꾸고 남은 나머지 rest를 구하면서 n의 값이 a의 값보다 낮아질 때까지 반복하면 되는 문제입니다. 그래서 반복을 하는 while문 안에 조건을 넣고 rest값과 rest값 + 빈병을 반납해서 받은 병들의 합을 구하도록 코드를 구성해서 문제를 풀어보았습니다.

 

1. 다른 사람 코드

class Solution {
    fun solution(a: Int, b: Int, n: Int): Int {
        return (if (n > b) n - b else 0) / (a - b) * b
    }
}

저의 코드와 다르게 1줄로 해결된 코드가 있어서 공부용으로 가져와봤습니다. 아직까지 코드를 완벽하게 이해가 안 되어서 향후 좀 더 공부해 봐야겠습니다. 그리고 여기서 n이 b보다 크다는 제한 사항이 있기 때문에 if문은 없어도 될 것 같습니다.

---

Performance

1. 내가 작성한 코드	2. 다른 사람 코드