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

지은·2023년 3월 23일
0

Algorithm

목록 보기
17/33

문제

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

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

콜라 빈 병 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 < an ≤ 1,000,000
정답은 항상 int 범위를 넘지 않게 주어집니다.

입출력 예

abnresult
212019
31209

입출력 예 설명

입출력 예 #1

본문에서 설명한 예시입니다.

입출력 예 #2

빈 병 20개 중 18개를 마트에 가져가서, 6병의 콜라를 받습니다. 이때 상빈이가 가지고 있는 콜라 병의 수는 8(20 – 18 + 6 = 8)개 입니다.
빈 병 8개 중 6개를 마트에 가져가서, 2병의 콜라를 받습니다. 이때 상빈이가 가지고 있는 콜라 병의 수는 4(8 – 6 + 2 = 4)개 입니다.
빈 병 4 개중 3개를 마트에 가져가서, 1병의 콜라를 받습니다. 이때 상빈이가 가지고 있는 콜라 병의 수는 2(4 – 3 + 1 = 2)개 입니다.
3번의 교환 동안 상빈이는 9(6 + 2 + 1 = 9)병의 콜라를 받았습니다.


나의 풀이

마트로부터 받은 콜라 병의 수를 마지막에 리턴해야 하기 때문에, received라는 변수로 선언하고 0을 할당했다.

현재 가지고 있는 콜라 병의 수(n)가 마트에 줘야 하는 콜라 병의 수(a) 보다 크거나 같을 동안 아래의 반복문을 반복한다.

  1. 마트에 줄 수 있는 빈 병 묶음을 give라는 변수로 선언하고, na로 나눈 값을 내림 처리해 할당한다.
    e.g. n = 20, a = 3 이라면 give는 20 / 3 = 6.xxx이다.
    즉, 마트에게 6 묶음의 빈 병을 주고, 묶음당 b개의 새로운 콜라를 받을 수 있다.
  2. 현재 가지고 있는 콜라 병의 수(n)에서 마트에 준 콜라병의 수 만큼 뺀다.
    n = n - (6 * 3);
    마트에 18개의 빈 병을 줬으니, 현재 가지고 있는 콜라 병의 수는 2가 된다.
  3. 마트에게 준 빈 병의 묶음 수 만큼 b개의 새로운 콜라를 받는다.
    n = n + (6 * 1);
  4. received 변수에 받은 콜라의 수를 더한다.
    received = received + (6 * 1);

반복문이 종료되고 received를 반환한다.

function solution(a, b, n) {
    let received = 0;
    while (n >= a) {
        let give = Math.floor(n / a);
        n -= a * give;
        n += give * b;
        received += give * b;
    }
    return received;
}
profile
블로그 이전 -> https://janechun.tistory.com

2개의 댓글

comment-user-thumbnail
2023년 3월 26일

알고리즘 화이팅 !

1개의 답글