백준 알고리즘 28701번

03 Oct 2023 in Dev / Python on Python

• 백준 알고리즘 28701번 세제곱의 합
  • 문제
  • 입력
  • 출력
  • 예제
  • 풀이

백준 알고리즘 28701번 세제곱의 합

문제

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

• (1 + 2 + … + N)² = 1³ + 2³ + … + N³

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

입력

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

출력

세 줄을 출력하세요.

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

예제

입력	출력
5	15 225 225
100	5050 25502500 25502500

풀이

n = int(input())
total = 0
cubes = 0

for i in range(1, n+1):
  total += i
  cubes += i ** 3
print(total)
print(total ** 2)
print(cubes)