結果

問題 No.3505 Sum of Prod of Root
コンテスト
ユーザー Kude
提出日時 2026-04-18 02:53:20
言語 C++23
(gcc 15.2.0 + boost 1.89.0)
コンパイル:
g++-15 -O2 -lm -std=c++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 216 ms / 2,000 ms
コード長 1,908 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,863 ms
コンパイル使用メモリ 361,656 KB
実行使用メモリ 19,956 KB
最終ジャッジ日時 2026-04-18 02:53:26
合計ジャッジ時間 4,697 ms
ジャッジサーバーID
（参考情報） 		judge1_0 / judge2_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 13
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include<bits/stdc++.h>
namespace {
#pragma GCC diagnostic ignored "-Wunused-function"
#include<atcoder/all>
#pragma GCC diagnostic warning "-Wunused-function"
using namespace std;
using namespace atcoder;
#define rep(i,n) for(int i = 0; i < (int)(n); i++)
#define rrep(i,n) for(int i = (int)(n) - 1; i >= 0; i--)
#define all(x) begin(x), end(x)
#define rall(x) rbegin(x), rend(x)
template<class T> bool chmax(T& a, const T& b) { if (a < b) { a = b; return true; } else return false; }
template<class T> bool chmin(T& a, const T& b) { if (b < a) { a = b; return true; } else return false; }
using ll = long long;
using P = pair<int,int>;
using VI = vector<int>;
using VVI = vector<VI>;
using VL = vector<ll>;
using VVL = vector<VL>;
using mint = modint998244353;

} int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);
  ll n;
  cin >> n;
  struct Q {
    ll t;
    mint c;
  };
  vector<Q> evs;
  rep(k, 60) if (k >= 3) {
    for (int i = 2;; i++) {
      ll v = 1;
      rep(_, k) if (__builtin_mul_overflow(v, i, &v)) v = n + 1;
      if (v > n) break;
      evs.emplace_back(v, mint(i) / mint(i-1));
    }
  }
  ranges::sort(evs, {}, &Q::t);
  evs.emplace_back(n + 1, 0);
  ll t = 1;
  mint c = 1;
  mint ans;
  static const mint inv2 = mint(2).inv();
  auto f1 = [](ll l, ll r) {
    return mint(l + r - 1) * (r - l) * inv2;
  };
  for (auto [nt, nc] : evs) {
    ll l = t, r = nt;
    ll sql = sqrt(l);
    ll nl = (sql + 1) * (sql + 1);
    if (nl >= r) {
      ans += c * sql * f1(l, r);
    } else {
      ans += c * sql * f1(l, nl);
      sql++;
      ll sqr = sqrt(r);
      ll nr = sqr * sqr;
      ans += c * sqr * f1(nr, r);
      mint L = sql, R = sqr;
      static const mint inv20 = mint(20).inv();
      ans += c * inv20 * ((R - 1) * R * (R + 1) * (R * (8 * R - 5) - 2) - (L - 1) * L * (L + 1) * (L * (8 * L - 5) - 2));
    }
    t = nt;
    c *= nc;
  }
  cout << ans.val() << '\n';
}
0