코딩테스트/프로그래머스

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

박소민 2022. 3. 9. 16:15
부족한 금액 계산하기
 

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

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

programmers.co.kr

 

  • 내 풀이
def solution(price, money, count):
    answer=0
    for i in range(count):
        answer+=price*(i+1)
    if money-answer>=0:
        return 0
    answer-=money
    return answer
print(solution(3,20,4))
#결과
10

 

 

  • 다른 사람 풀이

- 점차 n배수씩 되는 값들의 합 : n(n+1)//2 로 이용

def solution(price, money,count):
  return max(0, price*(count+1)*count//2-money)