level1_부족한 금액 계산하기

2024. 2. 3. 22:14파이썬_알고리즘(코딩테스트)/level1

 

 

 

내가 제출한 코드

def solution(price, money, count):
    answer = 0
    for i in range(1, count +1):
        answer += price * i
    if answer > money:
        return answer - money
    else:
        return 0

 

다른 사람이 제출한 코드

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

 

등차수열 합 공식은 진짜 많이 쓰이는거 같다.

- price*(count+1)*count//2

자세한 설명은 level1_두 정수 사이의 합 게시물에 자세하게 적어뒀다.

 

- max(0, 계산된 총 비용 - money)

사용자가 충분한 돈을 가지고 있으면 0을 반환하고, 그렇지 않다면 그 차액이 반환된다.