[BOJ][Python] 백준 28701번: 세제곱의 합 (Bronze V)

https://www.acmicpc.net/problem/28701

 

문제

은하는 수업 때 $1$부터 $N$까지 수의 합과 $1$부터 $N$까지 수의 세제곱의 합과 관련된 다음 공식을 배웠습니다.
믿을 수 없었던 은하는 직접 코딩을 해서 검증해 보기로 했습니다. $1$부터 $N$까지 수의 합과 그 수를 제곱한 수, 또 $1$부터 $N$까지 수의 세제곱의 합을 차례대로 출력하세요.

입력

첫 줄에 문제의 정수 $N$이 주어집니다. $(5 \le N \le 100)$

출력

세 줄을 출력하세요.

 

풀이

 

n = int(input())

a = b = c = 0

for i in range(1, n+1):
    a += i
    c += i*i*i
b = a*a

print(a)
print(b)
print(c)