[프로그래머스] 부족한 금액 계산하기 (Python)

문제 링크 (Level 1)

https://programmers.co.kr/learn/courses/30/lessons/82612?language=python3 

코딩테스트 연습 - 부족한 금액 계산하기

 

나의 코드 (Python)

def solution(price, money, count):
    answer = 0
    for i in range(1, count+1):
        answer += price * i
    
    return answer - money if answer > money else 0

 

놀이기구를 탈 때마다 가격이 횟수의 곱만큼 뛰게 됩니다.

이때 결과가 가진 돈보다 많다면 answer - money 적다면 0을 결과로 줍니다.