結果

問題 No.2891 Mint
コンテスト
ユーザー SnowBeenDiding
提出日時 2024-09-13 22:15:57
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,093 ms / 2,000 ms
コード長 1,099 bytes
コンパイル時間 5,527 ms
コンパイル使用メモリ 311,548 KB
実行使用メモリ 97,408 KB
最終ジャッジ日時 2024-09-13 22:16:32
合計ジャッジ時間 29,894 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge6
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 54
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <atcoder/all>
#include <bits/stdc++.h>
#define rep(i, a, b) for (ll i = (ll)(a); i < (ll)(b); i++)
using namespace atcoder;
using namespace std;

typedef long long ll;
using mint = modint998244353;

mint greedy(ll n, ll m) {
    mint ans = 0;
    rep(k, 1, n + 1) { ans += m % k; }
    return ans;
}

mint f(ll n, ll m) {
    mint ans = 0;
    set<ll> st;
    for (ll i = 1; i * i <= m; i++) {
        st.insert(i);
        st.insert(m / i);
    }
    for (ll i : st) {
        ll xl = m / (i + 1) + 1;
        ll xr = m / i;
        xr = min(xr, n);
        if (xl <= xr) {
            ans += (mint)i * (xl + xr) * (xr - xl + 1) / 2;
        }
    }
    return ans;
}

mint solve(ll n, ll m) {
    mint ans = (mint)m * n;
    ans -= f(n, m);
    return ans;
}

int main() {
    ll n, m;
    cin >> n >> m;
    cout << solve(n, m).val() << endl;
    // rep(i, 1, 100) rep(j, 1, 100) {
    //     if (solve(i, j).val() != greedy(i, j).val()) {
    //         cout << i << " " << j << " " << solve(i, j).val() << " "
    //              << greedy(i, j).val() << endl;
    //     }
    // }
}
0