結果

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

ソースコード

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;
    for(auto &&c : n){
        ans *= mod;
        ans += dp2[L][0][c - '0'] + dp2[L][0][c - '0'];
    }
    cout << ans.val()  << '\n';
}
0