[위클리 챌린지 1주차] - 부족한 금액 계산하기 (파이썬)

프로그래머스의 위클리 챌린지 1주차를 풀었습니다.

문제 링크: https://programmers.co.kr/learn/courses/30/lessons/82612

코딩테스트 연습 - 1주차_부족한 금액 계산하기

 

풀이

놀이기구를 count번 타게 되었을때 필요한 금액

예시) count = 5, price = 100

• 100*1+100*2+100*3+100*4+100*5 = 1500

for문을 이용해서 1부터 count까지 price를 곱한 금액을 총 비용으로 변수에 할당

예산인 money보다 총 비용이 작으면 money에서 총 비용을 뺀 값을 리턴

가지고 있는 예산부터 총 비용이 많이 발생하면 0을 리턴

 

코드

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