[프로그래머스] 부족한 금액 계산하기 Lv.1 - [파이썬/python]
프로그래머스에서 출제한 82612번 문제인 "부족한 금액 계산하기" Lv.1을 파이썬으로 풀어보겠습니다.
💻 문제 설명
새로 생긴 놀이기구는 인기가 매우 많아 줄이 끊이질 않습니다.
이 놀이기구의 원래 이용료는 price 원 인데, 놀이기구를 N 번 째 이용한다면 원래 이용료의 N배를 받기로 하였습니다.
즉, 처음 이용료가 100이었다면 2번째에는 200, 3번째에는 300으로 요금이 인상됩니다.
놀이기구를 count 번 타게 되면 현재 자신이 가지고 있는 금액에서 얼마가 모자라는지를 반환(return)하는 문제입니다.
단, 금액이 부족하지 않으면 0을 반환(return)하세요.
🚨 제한사항
- 놀이기구의 이용료 price : 1 ≤ price ≤ 2,500, price 는 자연수
- 처음 가지고 있던 금액 money : 1 ≤ money ≤ 1,000,000,000, money 는 자연수
- 놀이기구의 이용 횟수 count : 1 ≤ count ≤ 2,500, count 는 자연수
🌈 힌트
더보기
- 등차수열 공식 활용
놀이기구를 연속해서 탈 때 이용료가 등차수열로 증가합니다.
등차수열의 합 공식을 활용하여 총 비용을 계산할 수 있습니다. - 총 비용 계산
놀이기구를 count번 타기 위한 총 비용을 계산하고, 현재 가진 금액에서 이 값을 뺌으로써 모자라는 금액을 구할 수 있습니다. - 조건 확인
계산한 모자라는 금액이 음수가 아닌 경우, 모자라지 않고 충분한 금액을 가지고 있는 것입니다.
이 경우에는 0을 반환해야 합니다.
!!!정답 주의!!!
🌟 소스 코드
def solution(price, money, count):
# 등차수열의 합 공식을 사용하여 총 비용 계산
total_cost = price * (count * (count + 1) // 2)
# 금액이 부족하지 않은 경우 0을 반환
# 금액이 모자라는 경우 (total_cost - money) 반환
return max(0, total_cost - money)
🏳🌈 테스트 결과
728x90
'코딩테스트' 카테고리의 다른 글
| [프로그래머스] 대충 만든 자판 - Lv.1 (75) | 2023.10.02 |
|---|---|
| [프로그래머스] 콜라 문제 - Lv.1 (67) | 2023.10.02 |
| [프로그래머스] 덧칠하기 - Lv.1 (67) | 2023.10.01 |
| [프로그래머스] 크레인 인형뽑기 게임 - Lv.1 (63) | 2023.10.01 |
| [프로그래머스] 문자열 내 마음대로 정렬하기 - Lv.1 (56) | 2023.09.28 |