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

문제

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

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

콜라 빈 병 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 범위를 넘지 않게 주어집니다.

내가 생각한 풀이 방법

get_coke는 바꿔 받은 콜라 수

remain_coke는 바꾸러 가져가지 못하는 콜라 수

total_coke는 전체 콜라 수

n // a 의 결과를 get_coke와 total_coke에 저장한다.

만약 n % a가 0이 아니라면 그 결과를 remain_coke에 저장한다.

(이 부분까지 for문 while문에 넣는다면 get_coke = n으로 초기화)

 

여기부터 for문을 돌면서

while문을 돌면서 (while get_coke !=0)

만약 get_coke % a == 0일 경우

get_coke // a의 결과를 다시 get_coke에 저장하고 total_coke에 더해준다.

만약 get_coke % a가 0이 아니라면

(get_coke+remain_coke) // a의 결과를 get_coke에 저장하고 total_coke에 저장한다

(get_coke+remain_coke) % a 결과를 remain_coke에 저장한다.

 

 

코드로 구현

1차 시도

def solution(a, b, n):
    get_coke = n
    remain_coke = 0
    total_coke = 0
    
    while get_coke != 0:
        if get_coke % a == 0:
            get_coke = (get_coke // a)* b
            total_coke += get_coke
        else:
            get_coke = ((get_coke + remain_coke) // a) * b
            total_coke += get_coke
            remain_coke = (get_coke + remain_coke) % a
            
    return total_coke

- 기대 결과값보다 실행 결과 값이 1 적었다

- 중간에 로직이 좀 꼬인 것 같긴 한데,,

 

2차 시도

def solution(a, b, n):
    get_coke = n
    remain_coke = 0
    total_coke = 0
    
    while get_coke != 0:
        if get_coke % a == 0:
            get_coke = (get_coke // a)* b
            total_coke += get_coke
        else:
            remain_coke = (get_coke + remain_coke) % a
            get_coke = ((get_coke + remain_coke) // a) * b
            total_coke += get_coke
            
    return total_coke

- remain을 밑에 두면 이미 반환 완료된 뒤의(?) get_coke를 활용해 계산되는데 내가 의도했던건 그게 아니라서 앞부분으로 옮겼다.

- 그랬더니 이번에는 실행 결과값이 1 많게 나왔다..?

 

3차 시도

def solution(a, b, n):
    bottles = n
    get_coke = 0
    remain_coke = 0
    total_coke = 0
    
    while bottles > a:
        if bottles % a == 0:
            get_coke = (bottles // a)* b
            total_coke += get_coke
            bottles = get_coke
        else:
            remain_coke = bottles % a
            get_coke = (bottles // a) * b
            total_coke += get_coke
            bottels = get_coke + remain_coke
        if bottles == 1:
            if (bottles + remain_coke) % a == 0:
                total_coke += 1
            
    return total_coke

다시 정의해보자,,

get_coke는 받을 수 있는 콜라 수

remain_coke는 나눠지지 않아 남은 콜라 수

bottles는 가지고 있는 콜라 수

total_coke는 전체 받을 수 있는 콜라수

- 시간 초과로 결과도 못 보고 실패

 

4차 시도

 

 

 

 

-----------------------------------------------------------

def solution(a, b, n):
    total_coke = 0  # 상빈이가 전체적으로 받을 수 있는 콜라의 양

    while n >= a:  # 빈 병이 a개 이상인 동안
        # a개의 빈 병으로 b병의 콜라를 받을 수 있음
        exchanged_coke = (n // a) * b
        total_coke += exchanged_coke

        # 남아있는 빈 병 + 새롭게 마시게 될 콜라 병의 개수로 n 업데이트
        n = n - (n // a) * a + exchanged_coke

    return total_coke

 

다른 사람 풀이