[Baekjoon/Python3] 1834번 나머지와 몫이 같은 수

HooNeee 2020. 12. 3. 01:04

1834번: 나머지와 몫이 같은 수

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

www.acmicpc.net

n = int(input())
total = 0
for i in range(n + 1, pow(n, 2), n + 1):    # pattern
    if i % n == i // n:
        total += i
print(total)