(Python) 콜라 (프로그래머스 Lv.1)

콜라 (프로그래머스 Lv.1)

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

프로그래머스

 

내 코드

def solution(a, b, n):
    answer = 0
    rest = 0
    while(n>=a):
        rest = n%a
        n  =  n//a*b
        answer += n
        n += rest
    return answer

 

rest라는 남는 병을 저장해주는 변수를 따로 지정해서 모드를 짜봤다. rest가  나눌 거 다 나누고 마지막에 들어가야 한다.