[Python] 위클리 챌린지 : 부족한 금액 계산하기

https://programmers.co.kr/learn/courses/30/lessons/82612

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

 

💡 문제

 

---

새로 생긴 놀이기구는 인기가 매우 많아 줄이 끊이질 않습니다. 이 놀이기구의 원래 이용료는 price원인데, 놀이기구를 N번째 이용하다면 원래 이용료의 N배를 받기로 하였습니다. 놀이기구를 count번 타게 되면 현재 자신이 가지고 있는 금액에서 얼마가 모자라는지를 return 하도록 solution 함수를 완성하세요. 단, 금액이 부족하지 않으면 0을 return 하세요.

 

 

 

[입출력 예]

 

#1

이용료가 3인 놀이기구를 4번 타게 되면 총 이용료는 3+6+9+12 = 30이므로, 기존 가지고 있던 금액 20보다 10 모자라다.

 

 

 

 

[ Python Code 👩🏻‍💻]

def solution(price, money, count):
    total_price = [price*i for i in range(1,count+1)]
    if(sum(total_price)<=money):
        return 0
    else:
        return sum(total_price) - money

 

Code Review >

1. 놀이기구 이용료를 계산하여 저장한다.

2. 만약 기존에 가지고 있던 금액보다 총 이용료가 적다면 0을 return 한다.

3. 그렇지 않으면 총이용료에서 가지고 있는 금액을 빼준다.

 

 

 

 

 

더욱더 간결하게 구현한 코드는 다음과 같다😊

 

 

 

[Python Code👩🏻‍💻]

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

 

Code Review >

1. 놀이기구 이용료를 계산하여 저장한다.

2. 0과 두 값을 뺀 것 중 max 값을 return 한다. (만약 가지고 있던 금액이 총이용료보다 크면 계산된 값을 음수이므로 0을 return)