結果
問題 | No.2127 Mod, Sum, Sum, Mod |
ユーザー |
![]() |
提出日時 | 2022-11-18 22:02:33 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 5 ms / 2,000 ms |
コード長 | 1,253 bytes |
コンパイル時間 | 1,859 ms |
コンパイル使用メモリ | 173,444 KB |
実行使用メモリ | 6,948 KB |
最終ジャッジ日時 | 2024-09-20 02:24:21 |
合計ジャッジ時間 | 2,411 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 27 |
ソースコード
#include <bits/stdc++.h>using namespace std;const int MOD = 998244353;const int ONE_SIXTH = 166374059;vector<pair<int, pair<int, int>>> quotient_ranges(int N){vector<pair<int, pair<int, int>>> ans;for (int i = 1; i * i <= N; i++){ans.push_back(make_pair(N / i, make_pair(i, i)));}for (int i = N / ((int) sqrt(N) + 1); i >= 1; i--){ans.push_back(make_pair(i, make_pair(N / (i + 1) + 1, N / i)));}return ans;}long long sum1(long long x){return x * (x + 1) / 2 % MOD;}long long sum2(long long x){return x * (x + 1) % MOD * (x * 2 + 1) % MOD * ONE_SIXTH % MOD;}int main(){int N, M;cin >> N >> M;long long ans = (long long) N * (N + 1) / 2 % MOD * M % MOD;vector<pair<int, pair<int, int>>> Q = quotient_ranges(N);int cnt = Q.size();long long ans2 = 0;for (int i = 0; i < cnt; i++){int q = Q[i].first;int mn = Q[i].second.first;int mx = Q[i].second.second;mx = min(mx, M);if (mn <= mx){long long S1 = (long long) (N + 1) * (sum1(mx) - sum1(mn - 1) + MOD) % MOD * q % MOD;long long S2 = (long long) q * (q + 1) / 2 % MOD;S2 *= (sum2(mx) - sum2(mn - 1) + MOD) % MOD;S2 %= MOD;ans += MOD - S1 + S2;}}ans %= MOD;cout << ans << endl;}