[파이썬/python] 백준 1834번 나머지와 몫이 같은 수

 

 

 

문제

 

N으로 나누었을 때 나머지와 몫이 같은 모든 자연수의 합을 구하는 프로그램을 작성하시오. 예를 들어 N=3일 때, 나머지와 몫이 모두 같은 자연수는 4와 8 두 개가 있으므로, 그 합은 12이다.

 

 

 

 

 

 

 

입력

 

첫째 줄에 2,000,000 이하의 자연수 N이 주어진다.

ex)

3

2000000

 

 

 

 

 

출력

 

첫 줄에 구하고자 하는 수를 출력한다.

ex)

12

3999999999999000000

 

---

---

 

코드

 

import sys
import math

n=int(sys.stdin.readline())

sum=0

for i in range(n+1, int(math.pow(n,2)), n+1):
    sum=sum+i

print(sum)