[Programers][Python] 부족한 금액 계산하기

보자마자 price(1~count) 가 생각나서 제출. 했으나 제출 누르자마자 음수인 경우 예외처리 안해줬다는 생각이 나서

결과보니 아니나다를까 testcase 하나 오답.. 음수처리 해주었더니 통과

 

def solution(price, money, count):
    answer = price * count * (count + 1) // 2 - money
    if answer < 0:
        answer = 0
    return answer