def solution(price, money, count): answer = 0 for m in range(1, count+1): answer += price * m if answer <= money: answer = 0 else: answer = answer - money return answer
너무 기초 문제라서 따로 작성할 것이 없다..