[알고리즘] 프로그래머스 - 부족한 금액 계산하기

부족한 금액 계산하기

프로그래머스

 

1. 구해야 하는 것
    # 금액이 얼마나 모자란지 (아니면 0 반환)

2. 1을 구하기 위해 필요한 것
    # 놀이기구를 count번 탔을 때 비용의 합
        # for i in range(1, count+1)
        # cost에 i*price를 계속 더해주기
    # money에서 cost를 뺀다

 

 

1부터 count까지 i와 price를 곱해 해당 회차의 가격을 구하고, 구한 가격을 모두 더해준다.

그 후 가진 돈에서 빼서 얼마가 모자란지 계산, 부족한지 여부에 따라 0또는 answer를 리턴

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