結果

問題 No.2891 Mint
コンテスト
ユーザー t98slider
提出日時 2024-09-14 19:01:53
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 52 ms / 2,000 ms
コード長 906 bytes
コンパイル時間 4,083 ms
コンパイル使用メモリ 251,468 KB
最終ジャッジ日時 2025-02-24 08:38:40
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 54
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using ll = long long;
using mint = atcoder::modint998244353;

template<class T> istream& operator >> (istream& is, vector<T>& vec) {
    for(T& x : vec) is >> x;
    return is;
}

//等差数列の和を求める(初項,項数,公差)
mint arith_sum1(long long fst, long long n, long long d){
    return mint(n) * (fst + (n - 1) * mint(d) * 499122177);
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    ll n, m;
    cin >> n >> m;
    mint ans;
    if(n > m) ans += mint(n - m) * m;
    n = min(n, m - 1);
    ll cur = 1;
    while(cur <= n && cur * cur <= m){
        ans += m % cur;
        cur++;
    }
    while(cur <= n){
        ll d = m / cur;
        ll nxt = m / d + 1;
        ll dn = min(n + 1, nxt) - cur;
        ans += arith_sum1(m % cur, dn, -d);
        cur = nxt;
    }
    cout << ans.val() << '\n';
}
0