結果

問題 No.3118 Increment or Multiply
コンテスト
ユーザー 入野拳樹
提出日時 2025-04-20 13:03:38
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 915 bytes
コンパイル時間 2,155 ms
コンパイル使用メモリ 198,920 KB
実行使用メモリ 814,080 KB
最終ジャッジ日時 2025-04-20 13:03:43
合計ジャッジ時間 4,497 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 1
other AC * 5 MLE * 1 -- * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

const ll MOD = 998244353;

ll modmul(ll a, ll b) {
    return (a % MOD) * (b % MOD) % MOD;
}

ll modadd(ll a, ll b) {
    return (a + b) % MOD;
}

ll dfs(ll n, ll a, map<ll, ll>& memo) {
    if (memo.count(n)) return memo[n];
    if (n == 1) return 0;

    ll res = 0;
    if (a == 1) {
        res = modmul(n % MOD, (n - 1) % MOD);
        res = modmul(res, (MOD + 1) / 2); 
        return memo[n] = res;
    }

    ll next = (n - 1) / a * a;
    ll count = n - next;
    
    ll add_sum = modmul(count % MOD, ((count - 1 + MOD) % MOD) * ((MOD + 1) / 2) % MOD);
    add_sum = modadd(add_sum, modmul(count % MOD, dfs(next, a, memo)));

    return memo[n] = add_sum % MOD;
}

int main() {
    int T;
    cin >> T;
    while (T--) {
        ll N, A;
        cin >> N >> A;
        map<ll, ll> memo;
        cout << dfs(N, A, memo) << endl;
    }
}
0