백준 28701 세제곱의 합(python)-[bronze(5)]

문제-https://www.acmicpc.net/problem/28701

28701번: 세제곱의 합

수의 합의 제곱과 수의 세제곱의 합이 같다는 것을 확인해보는 코드 이므로, 더 짧게 할 수 있었음에도 문제 취지에 맞게 풀어 보았다.

 

코드

---

N = int(input())
S = 0
SQ = 0
for i in range(1, N+1):
    S += i
    SQ = SQ + i**3
    
print(S)
print(S**2)
print(SQ)

입력 수 - N

모든 수의 합 - S

세제곱의 합 - SQ

 

반복문 범위 입력시 0을 빼서 불필요한 시간을 줄였다. (사실 시간적으로 큰 차이는 없다.)

 

결과

input: 100
ouput: 5050
ouput: 25502500
ouput: 25502500