프로그래머스 Lv1 부족한 금액 계산하기 (python)

부족한 금액 계산하기

풀이

한 번 탈 때마다 price 요금을 더 내야 하는 이상한 놀이기구이다.
얼마나 인기가 많으면 이렇게 배짱장사를 할까...

간단하게 풀어진다.

1. 반복문을 돌리면서 count만큼 탔을 때의 비용을 계산한다.
2. 내가 들고 있는 돈과 비용을 계산하고 그에 따른 답을 반환한다.

# price원 -> N번째시 price * N
def solution(price, money, count):
    i = 0
    whole_price = 0
    while i< count:
        i += 1
        whole_price += price*i
    if whole_price > money:
        return whole_price - money
    else:
        return 0

다른 사람 풀이

max()함수를 사용하고 또한, 수학적으로 값을 계산해서 아래처럼 풀었다.
역시 똑똑한 사람들이 많다.

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