[Level1] 콜라 문제 (C++/Python)

C++)

#include <string>
#include <vector>
 
using namespace std;
 
int solution(int a, int b, int n) {
    int answer = 0;
    
    while(n >= a)
    {
        answer += ((n/a) * b);
        n = (((n/a) * b) + n%a);
    }
    
    return answer;
}
 
// (n/a * b) == 교환한 콜라병
// n%a == a개를 채우지 못해 교환하지 못한 콜라병

 

 

 

Python)

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