-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathSumOfDivisors.cpp
More file actions
47 lines (37 loc) · 865 Bytes
/
SumOfDivisors.cpp
File metadata and controls
47 lines (37 loc) · 865 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
//Sum of Divisors - https://cses.fi/problemset/task/1082
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MOD = 1e9 + 7;
const ll INF = 1e18;
const ll TWO_INVERSE = 500000004;
void solve() {
ll n;
cin >> n;
ll sq = (ll) sqrtl(n);
ll ans = 0;
for (ll d = 1; d <= n / (sq + 1); d++) {
ans += ((d * (n / d)) % MOD);
ans %= MOD;
}
auto f = [&](ll x) -> ll {
x %= MOD;
return (((x * (x + 1)) % MOD) * TWO_INVERSE) % MOD;
};
for (ll d = 1; d * d <= n; d++) {
ll diff = (f(n / d) - f(n / (d + 1)) + MOD) % MOD;
ans += (d * diff) % MOD;
ans %= MOD;
}
cout << ans << endl;
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(nullptr);
int T = 1;
// cin >> T;
while (T--) {
solve();
}
return 0;
}