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

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

프로그래머스

문제 예시에 나와있는 연산을 그대로 활용하면 되는 문제

 

풀이 :

1. 무한루프를 돌면서 n < a일 경우 반복문을 종료하도록 종료 조건 설정

2. 매 반복 시점마다 answer에 n(현재 병의 수)을 a로 나눈 몫에 b를 곱한 값을 더해줌

3. 매 반복 시점마다 n = n - (n에서 n을 a로 나눈 나머지를 빼준 값) + (빈 병을 가져가서 받은 병 수)로 최신화 

 

코드 :

def solution(a, b, n):
    answer = 0
    while True:
        if n < a:
            break
        answer += (n // a)*b
        n = n - (n - n % a) + (n // a)*b

    return answer