結果

問題 No.3089 Base M Numbers, But Only 0~9
コンテスト
ユーザー t98slider
提出日時 2025-04-04 23:42:41
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,517 bytes
コンパイル時間 5,264 ms
コンパイル使用メモリ 257,540 KB
実行使用メモリ 7,844 KB
最終ジャッジ日時 2025-04-04 23:42:47
合計ジャッジ時間 5,431 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 3
other AC * 6 WA * 13
権限があれば一括ダウンロードができます

ソースコード

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;
    bool lz = true;
    for(int i = 0; i < c; i++){
        if(n[i] == '0'){
            L[i + 1] = L[i] * (dp[d][0][n[i] - '0'] - lz);
        }else{
            lz = false;
            L[i + 1] = L[i] * dp[d][0][n[i] - '0'];
        }
    }
    if(lz){
        cout << "0\n";
        return 0;
    }
    for(int i = c - 1; i >= 0; i--){
        R[i] = R[i + 1] * dp[d][0][n[i] - '0'];
    }
    mint ans, mod = m;
    for(int i = 0; i < c; i++){
        ans *= mod;
        ans += dp2[d][0][n[i] - '0'] * L[i];
    }
    cout << ans.val()  << '\n';
}
0