結果

問題 No.2891 Mint
コンテスト
ユーザー kusaf_
提出日時 2024-11-02 02:54:37
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 135 ms / 2,000 ms
コード長 730 bytes
コンパイル時間 3,487 ms
コンパイル使用メモリ 251,820 KB
実行使用メモリ 54,344 KB
最終ジャッジ日時 2024-11-02 02:54:46
合計ジャッジ時間 8,587 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 54
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

vector<tuple<ll, ll, ll>> EnumFloor(ll n) {
  vector<tuple<ll, ll, ll>> ret;
  ll l = 1;
  while(l <= n) {
    ll q = n / l, r = n / q;
    ret.push_back({l, r, q});
    l = r + 1;
  }
  return ret;
}

int main() {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);

  ll N, M;
  cin >> N >> M;

  const mint inv = mint(2).inv();

  mint ans = mint(N) * M;

  for(auto &[l, r, q] : EnumFloor(M)) {
    bool fin = false;
    if(N <= r) {
      r = N;
      fin = true;
    }
    ans -= mint(l + r) * (r - l + 1) * inv * q;
    if(fin) { break; }
  }

  cout << ans.val() << "\n";
}
0