Programmers - 부족한 금액 계산하기(Python)

문제

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

제한사항

  • 3 ≤ n ≤ 1,000,000

입출력 예

pricemoneycountresult
320410

입출력 예 설명

  • 입출력 예 #1
    • 이용금액이 3인 놀이기구를 4번 타고 싶은 고객이 현재 가진 금액이 20이라면, 총 필요한 놀이기구의 이용 금액은 30 (= 3+6+9+12) 이 되어 10만큼 부족하므로 10을 return 합니다.

✍ 코드

def solution(price, money, count):
    answer = 0 

    # (1)
    sum_money = 0

    # (2)
    for i in range(1, count+1):
        sum_money += price * i

    # (3)
    if sum_money >= money :
        answer = sum_money - money

    return answer

price = 3
money = 20
count = 4
print(solution(price, money, count)) # 출력 예 : 10 
  • (1) : 타는 횟수에 따른 금액을 할당하기 위한 변수 sum_money
  • (2) : for문을 통해 sum_money 계산하기
  • (3) : sum_money가 money보다 크거나 같으면 필요한 이용 금액 산출하기

📝 결과

😃 느낀점

  • 문제 지문이 길어서 순간 당황했지만, 지문을 다 읽고나시 그렇게 어려운 문제라고 판단되지 않았다. 따로 메소드나 라이브러리 등을 사용하지 않고도 풀었어서 괜찮았던 것 같다.

👍 Irish의 모든 코드 보기

-> Irish Github

좋은 웹페이지 즐겨찾기