結果
問題 | 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; }