부족한 금액 계산하기 - Python

1. 풀이 코드

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


# Test Cases
print(solution(3, 20, 4))
print(solution(3, 20, 1))

  위와 같이 등차수열의 합으로 계산하였습니다.

 

2. 다른 사람의 풀이 코드

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

  저는 음수인 경우에 대해서 변수 선언을 한 뒤에 if()문을 사용했는데 max()를 사용하면 이렇게 깔끔하게 정리가 되네요. 대단합니다.

 

---

• 프로그래머스: https://school.programmers.co.kr/learn/courses/30/lessons/82612

프로그래머스