[프로그래머스] 부족한 금액 계산하기 - 파이썬

위클리 문제라고해서 풀어봤다

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


    return answer

---

그냥 반복문을 count번만큼 돌면서 가격을 곱한것을 다 더해주었다가 그 값이 가지고 있는 돈money보다 커지면 money를 빼서 리턴시킨다.

만약 res가 money보다 작을 수도 있으니까 그럴 땐 돈이 부족하지 않다는 의미이다. 그래서 리턴 0