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

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

 

내 풀이

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

 

다른 방법

def solution(price, money, count):
    return ((count+1)*count/2)*price-money if  ((count+1)*count/2)*price-money>0 else 0