백준 알고리즘 1834번 나머지와 몫이 같은 수(python)

입력이 n이라고 하자. n보다 작은 수는 몫이 0이지만 나머지는 0일 수가 없기 때문에 제외하고,

n의 제곱보다 큰 수는 몫이 n 보다 커져서 나머지와 같을 수가 없기 때문에 제외한다.

남은 범위에서 나머지와 몫이 같은 수를 찾으면 된다.

EX) n = 3 -> 1, 2, 3은 몫은 0이지만 나머지가 0보다 크기 때문에 같을 수가 없고,

10, 11, 12...는 몫은 3보다 큰데 나머지는 3보다 클 수가 없다.

결과적으로 4(3+1) ~ 9(3**2)의 범위에서 나머지와 몫이 같은 수를 찾아주면 된다.

n = int(input())
s = 0
for i in range(n+1, n**2, n+1):
    s += i
print(s)