結果
問題 |
No.3118 Increment or Multiply
|
ユーザー |
|
提出日時 | 2025-04-12 15:26:23 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 17 ms / 2,000 ms |
コード長 | 1,844 bytes |
コンパイル時間 | 3,879 ms |
コンパイル使用メモリ | 278,476 KB |
実行使用メモリ | 7,844 KB |
最終ジャッジ日時 | 2025-04-12 16:43:30 |
合計ジャッジ時間 | 4,423 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 35 |
ソースコード
#include <bits/stdc++.h> using namespace std; using ll = long long; using pii = pair<int, int>; using pll = pair<ll, ll>; using vi = vector<int>; using vl = vector<ll>; #define rep3(i, a, b, c) for (ll i = (a); i < (b); i += (c)) #define rep2(i, a, b) rep3(i, a, b, 1) #define rep1(i, n) rep2(i, 0, n) #define rep0(n) rep1(aaaaa, n) #define ov4(a, b, c, d, name, ...) name #define rep(...) ov4(__VA_ARGS__, rep3, rep2, rep1, rep0)(__VA_ARGS__) #define per(i, a, b) for (ll i = (a) - 1; i >= (b); i--) #define fore(e, v) for (auto&& e : v) #define all(a) begin(a), end(a) #define sz(a) (int)(size(a)) #define lb(v, x) (lower_bound(all(v), x) - begin(v)) #define eb emplace_back template <typename T, typename S> bool chmin(T& a, const S& b) { return a > b ? a = b, 1 : 0; } template <typename T, typename S> bool chmax(T& a, const S& b) { return a < b ? a = b, 1 : 0; } const int INF = 1e9 + 100; const ll INFL = 3e18 + 100; #define i128 __int128_t struct _ { _() { cin.tie(0)->sync_with_stdio(0), cout.tie(0); } } __; const int MOD = 998244353; const int inv2 = (MOD + 1) / 2; void solve(); int main() { int T; cin >> T; while (T--) { solve(); } } void ch(ll& a, ll b) { a = (a + b) % MOD; } void solve() { ll N, A; cin >> N >> A; if (A == 1) { N %= MOD; ll ans = (N * (N - 1) / 2) % MOD; cout << ans << '\n'; return; } vl D; while (N > 0) { D.eb(N % A); N /= A; } ll ans = 0; ll cost = (accumulate(all(D), 0ll) + sz(D) - 1)%MOD; reverse(all(D)); ll t = 0; rep(i, sz(D)) { ll l = t + 1; t = (t * A + D[i]) % MOD; ll r = t, c = (r - l + 1 + MOD) % MOD; // cerr<<l<<' '<<r<<' '<<c<<' '; ch(ans, t * c - (l + r) * c % MOD * inv2); ch(cost, -(i != 0) - D[i]); // cerr<<cost<<endl; ch(ans, cost * c); } ch(ans, MOD); cout << ans << '\n'; }