본문 바로가기

알고리즘 삽질장

[프로그래머스] 부족한 금액 계산하기

반응형


문제설명

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

 

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

새로 생긴 놀이기구는 인기가 매우 많아 줄이 끊이질 않습니다. 이 놀이기구의 원래 이용료는 price원 인데, 놀이기구를 N 번 째 이용한다면 원래 이용료의 N배를 받기로 하였습니다. 즉, 처음 이

programmers.co.kr

사고과정

  • count 만큼 반복해서 money에서 price를 빼주되 count할 때마다 price에 1배씩 증가시켜 빼준다.
  • 결과 money값이 음수면 부족하다는 의미이므로 절댓값을 취해 부족한 금액을 반환하고 만약 양수라면 부족하지 않다는 것이므로 0을 반환하면 된다.

풀이

def solution(price, money, count):
    N = 1
    for _ in range(count):
        money -= price * N
        N += 1
    if money < 0:
        return abs(money)
    else:
        return 0
반응형