[구름 1단계] 태민이의 취미

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

 

알고리즘의 중요성을 맛볼 수 있는 문제, for문 사용시 time out으로 인해 풀 수 없음

# for문 사용시, 테스트케이스는 통과하는데 제출시에 통과 못함(time out 문제)

num = int(input())

count = 0
# 부피 = 가로 * 세로 * 높이
for i in range(1, num+1):
    count += i*i*i

# 1,000,000,007로 나눈 나머지 출력
print(count % 1000000007)

 

손으로 문제를 살짝 풀어보니 단순 수열문제였다.

2가지 방식으로 다시 풀어봤는데 해결법 2가 제일 보기 쉬고 간결한듯.

 

# 해결법 1
num = int(input())

ans = ((num ** 2) - ((num * (num-1)) // 2))**2
print((ans % 1000000007))

# 해결법 2
num = int(input())

ans = ((num * (num+1)) // 2) ** 2 
print((ans % 1000000007))