[Codeforces] 546A - Soldier and Bananas (Python)

문제

문제링크

바나나를 사기 위한 총 금액은 등차수열의 합과 같다.

총 금액에서 가지고 있는 돈을 빼서 부족한 금액을 구하라.

 

풀이

바나나의 개수는 w(w+1)//2와 같고 따라서 총 금액은 이 개수에 k를 곱한 값이다.

총 금액에서 가지고 있는 금액을 뺀 값을 rest에 저장한다.

rest가 0보다 크면 rest값을, 아니면 0을 출력한다.

 

코드

k,n,w=map(int, input().split())

tot=k*((w*(w+1))//2)
rest=tot-n
print(rest if rest>0 else 0)