링크: 문제 목록

부족한 금액 계산하기

문제

해결

프로그래머스를 정말 오랜만에 들어가보는지라 걱정했는데 다행히 아무거나 누른게 쉬운 문제였습니다. 정석적으로 for문과 if문을 사용할 수도 있습니다만, 저의 경우엔 자연수를 1부터 n까지 더하는 알고리즘과 비슷하다고 느꼈고, 고등학교 수학 시간에 배운 등차수열의 합 공식을 떠올렸습니다.

참고:등차수열의 합 공식

$\displaystyle{n*(n+1) \over 2}$

그러므로 변하지 않는 값인 price에 변하는 count변수를 위 수식에 대입해 코드를 짜주면 됩니다. 이때 공식에서 몫만 필요하므로 //연산자를 사용했습니다,

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