結果

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

ソースコード

diff # 

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

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    ll m;
    string n;
    cin >> m >> n;
    string s = to_string(m);
    const int L = s.size();
    vector dp(L + 1, vector(2, vector<mint>(10))), dp2(L + 1, vector(2, vector<mint>(10)));
    dp[0][1][0] = 1;
    for(int i = 0; i < L; 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;
                }
            }
        }
    }
    mint ans, mod = m, meth = 1;
    for(auto &&c : n){
        meth *= dp[L][0][c - '0'];
    }
    for(auto &&c : n){
        ans *= mod;
        ans += dp2[L][0][c - '0'] * (meth / dp[L][0][c - '0']);
    }
    cout << ans.val()  << '\n';
}
0