백준 파이썬 1964번 오각형, 오각형, 오각형... (Today I Learn 2023.11.10)

11/ 10 파이썬 공부
1. 백준 1964번 오각형, 오각형, 오각형...
1) 문제
오각형의 각 변에 아래 그림과 같이 점을 찍어 나간다. N단계에서 점의 개수는 모두 몇 개일까?

<그림>

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

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

>>>코드

n = int(input())
result = 5
for i in range(n-1):
    result += 3*i+7
print(result % 45678)

출력하는 것이 점의 개수가 아닌 점의 개수를 45678로 나눈 것이라는 것을 주의하자.

>>>문제 링크
https://www.acmicpc.net/problem/1964