콜라 문제 - Python

1. 풀이 코드

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


# Test Cases
print(solution(2, 1, 20))
print(solution(3, 1, 20))

  중간에 받는 병은 temp에 저장하여 차근차근 풀었습니다.

 

2. 다른 사람 풀이 코드

solution = lambda a, b, n: max(n - b, 0) // (a - b) * b

  수학적 계산을 잘 하네요.

 

---

• 프로그래머스: https://school.programmers.co.kr/learn/courses/30/lessons/132267

프로그래머스