[구름LEVEL1/수학] 태민이의 취미 - 파이썬

문제

https://level.goorm.io/exam/49094/%ED%83%9C%EB%AF%BC%EC%9D%B4%EC%9D%98-%EC%B7%A8%EB%AF%B8/quiz/1

구름LEVEL

 

풀이

정육면체의 부피는 한변의 길이가 n일 때 n*n*n 이다

길이가 1부터 n까지의 정육면체의 부피들의 합을 구하는 문제인데

반복문을 이용하여 n을 세제곱 하여 합을 구하게 된다면 시간초과로 오답을 출력한다.

 

반복문으로 구하지말고 세제곱의 합 공식을 이용하여 풀면 된다.

공식은 다음과 같다.

증명은 여기서 확인할 수 있다.

 

n = int(input())
result = ((n*(n+1))//2)**2
print(result % 1000000007)