백준 파이썬 28701번 세제곱의 합 (Today I Learn 2024.12.08)

백준 파이썬 공부 2024.12.08
28701번 세제곱의 합 (브론즈 5)

1. 문제
은하는 수업 때 1부터 N까지 수의 합과 1부터 N까지 수의 세제곱의 합과 
관련된 다음 공식을 배웠습니다.

 - (1 + 2 + ... + N)^2 = 1^3 + 2^3 + ... + N^3

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

2. 입력
첫 줄에 문제의 정수 N이 주어집니다. 
(5 <= N <= 100)

3. 출력
세 줄을 출력하세요.

- 첫 줄에는 1부터 N까지 수의 합 1+2+...+N을 출력하세요.
- 둘째 줄에는 1부터 N까지 수의 합을 제곱한 수 (1+2+...+N)^2을 출력하세요.
- 셋째 줄에는 1부터 N까지 수의 세제곱의 합 1^3+2^3+...+N^3을 출력하세요.

4. 코드

n = int(input())
# 합의 제곱
ans1 = 0
for i in range(n+1):
    ans1 += i
print(ans1)
print(ans1 ** 2)

# 세제곱들의 합
ans2 = 0
for i in range(n+1):
    ans2 += i ** 3
print(ans2)

5. 문제링크
https://www.acmicpc.net/problem/28701