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

📖 문제

https://programmers.co.kr/learn/courses/30/lessons/82612

💻 내 코드

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

    return answer

처음에 price*i를 해야하는데 price*count를 해버려서 결과 제대로 안나옴

💡 다른 사람 코드

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

• 수열 이용
• price*(1+2+3+...+count) = price*((count*(count+1))//2)
• 부족하지 않으면 0을 출력하기 위해 max 사용