[Python] 백준 1964번 오각형, 오각형, 오각형…

문제

오각형의 각 변에 아래 그림과 같이 점을 찍어 나간다. N단계에서 점의 개수는 모두 몇 개일까?

 

https://www.acmicpc.net/problem/1964

1964번: 오각형, 오각형, 오각형…

입력

첫째 줄에 N(1 ≤ N ≤ 10,000,000)이 주어진다.

출력

첫째 줄에 N단계에서 점의 개수를 45678로 나눈 나머지를 출력한다.

예제

# input
3
# output
22

# input
1
# output
5

# input
19
# output
590

문제 풀이

N = int(input())

ans = 5
gap = 7

for _ in range(1, N):
    ans += gap
    gap += 3
    
print(ans % 45678)

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,

혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.