[프로그래머스][level 1]부족한 금액 계산하기

부족한 금액 계산하기

-풀이

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

-풀이 설명

count 즉 놀이기구 탄 횟수만큼 반복문을 돌리고 tax변수에 가격*(횟수+1)를 추가로 더해준다. 만약 돈이 부족하지 않으면 0을 리턴하고 돈이 모자라면 tax에서 money를 빼주어 모자란 금액을 리턴해준다.