삽집하는 개발들/알고리즘

[49일차][Lv1][프로그래머스][132267]콜라 문제

악투 2023. 8. 29. 22:09
반응형
import math

def solution(a, b, n):
    result = 0    
    result_arr = []
    if n < a:
        return 0
    else:
        while n >= a:
            rest_val = n % a
            n = math.floor(n / a) * b
            result_arr.append(n)
            n += rest_val 
    
    return sum(result_arr)
반응형