[업무 지식]/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)