[BOJ/백준] 16673 - 고려대학교에는 공식 와인이 있다 (Python)

문제

문제링크

와인을 사기로 마음먹은 해를 0년 차라고 정의하고, n년 차에는 Kn+P(n**2) 만큼의 와인을 산다.

C년 동안 총 몇 병의 와인을 수집했는가?

 

풀이

y= Kn+P(n**2) 의 방정식을 return하는 함수를 작성한다.

각 해의 와인 수를 계산해 C년 동안의 총 합을 구하는 year_c 함수를 작성한다.

위 equation 함수를 불러와 i를 x값으로 사용해 i년의 와인 수를 구한다.

 

코드

def equation(x, K, P):
  return K*x+P*x*x

def year_c(C,K,P):
  tot=0
  for i in range(1, C+1):
    tot+=equation(i, K, P)
  return tot

C, K, P=map(int, input().split())
print(year_c(C, K, P))