結果

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

ソースコード

diff #
raw source code 

#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 L = s.size(), R = n.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;
                }
            }
        }
    }
    vector<mint> tmp(R);
    for(int i = 0; i < R; i++) tmp[i] = dp[L][0][n[i] - '0'];
    segtree<mint, op, e> seg(tmp);
    mint ans, mod = m;
    for(int i = 0; i < R; i++){
        ans *= mod;
        ans += dp2[L][0][n[i] - '0'] * seg.prod(0, i) * seg.prod(i + 1, R);
    }
    cout << ans.val()  << '\n';
}
0