結果

問題 No.3089 Base M Numbers, But Only 0~9
ユーザー t98slider
提出日時 2025-04-04 23:44:53
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 6 ms / 2,000 ms
コード長 1,324 bytes
コンパイル時間 4,540 ms
コンパイル使用メモリ 257,416 KB
実行使用メモリ 7,328 KB
最終ジャッジ日時 2025-04-04 23:45:00
合計ジャッジ時間 5,789 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

mint op(mint lhs, mint rhs){return lhs * rhs;};
mint e(){return 1;}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    ll m;
    string n;
    cin >> m >> n;
    string s = to_string(m);
    const int d = s.size(), c = n.size();
    vector dp(d + 1, vector(2, vector<mint>(10))), dp2(d + 1, vector(2, vector<mint>(10)));
    dp[0][1][0] = 1;
    for(int i = 0; i < d; i++){
        for(int j = 0; j < 2; j++){
            const int r = j ? s[i] - '0' : 9;
            for(int k = 0; k < 10; k++){
                for(int l = 0; l <= r; l++){
                    dp[i + 1][(j & (l == r))][l] += dp[i][j][k];
                    dp2[i + 1][(j & (l == r))][l] += l * dp[i][j][k] + dp2[i][j][k] * 10;
                }
            }
        }
    }
    vector<mint> R(c + 1), L(c + 1);
    L[0] = R[c] = 1;
    for(int i = 0; i < c; i++){
        L[i + 1] = L[i] * (dp[d][0][n[i] - '0'] - (i == 0 && n[i] == '0'));
        R.rbegin()[i + 1] = R.rbegin()[i] * dp[d][0][n.rbegin()[i] - '0'];
    }
    mint ans, mod = m;
    for(int i = 0; i < c; i++){
        ans *= mod;
        ans += dp2[d][0][n[i] - '0'] * L[i] * R[i + 1];
    }
    cout << ans.val()  << '\n';
}
0