[업무 지식]/Algorithm
[등차수열] 부족한 금액 계산하기
에디터 윤슬
2024. 12. 27. 09:25
링크
https://school.programmers.co.kr/learn/courses/30/lessons/82612
프로그래머스
SW개발자를 위한 평가, 교육, 채용까지 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프
programmers.co.kr
문제
정답 1
# 단순 반복문 사용
def solution(price, money, count):
total_cost = 0
for i in range(1, count + 1):
total_cost += price * i
return max(0, total_cost - money)
정답 2
# 등차수열 공식 활용
def solution(price, money, count):
total_cost = price * (count * (count + 1)) // 2
return max(0, total_cost - money)
정답 3
# 리스트 컴프리헨션 활용
def solution(price, money, count):
total_cost = sum(price * n for n in range(1, count+1))
return max(0, total_cost - money)