[Python] 백준 1834번 나머지와 몫이 같은 수

문제

N으로 나누었을 때 나머지와 몫이 같은 모든 자연수의 합을 구하는 프로그램을 작성하시오. 

예를 들어 N=3일 때, 나머지와 몫이 모두 같은 자연수는 4와 8 두 개가 있으므로, 그 합은 12이다.

 

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

입력

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

출력

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

예제

#input
3
#output
12

#input
2000000
#output
3999999999999000000

문제 풀이

# 입력받기
N = int(input())

# 정답 변수 초기화
answer = 0 

# 몫과 나머지가 같은 수 정답 변수에 더하기
for i in range(1, N):
    answer += N*i + i
        
print(answer)

---

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

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