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

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

 

boj.kr

풀이

반복문을 돌면서 조건에 맞는 수를 모두 더해주면 된다. 이 때 int 범위를 벗어날 수 있으므로 long long으로 처리해줘야 하는 것을 주의하자.

코드

C++

#include <iostream>
using namespace std;

int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);

    int n;
    cin >> n;
    long long sum = 0;
    for (long long i = 1; i < n; i++) {
        sum += (long long)(i * n + i);
    }
    cout << sum << '\n';
    return 0;
}

Python3

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