結果

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

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

#include <atcoder/all>
using namespace atcoder;

using mint = modint998244353;
using vm = vector<mint>;
using vvm = vector<vm>;
inline ostream &operator<<(ostream &os, const mint x)
{
    return os << x.val();
};
inline istream &operator>>(istream &is, mint &x)
{
    long long v;
    is >> v;
    x = v;
    return is;
};

struct Fast
{
    Fast()
    {
        std::cin.tie(nullptr);
        ios::sync_with_stdio(false);
        cout << setprecision(10);
    }
} fast;

#define all(a) (a).begin(), (a).end()
#define contains(a, x) ((a).find(x) != (a).end())
#define rep(i, a, b) for (int i = (a); i < (int)(b); i++)
#define rrep(i, a, b) for (int i = (int)(b) - 1; i >= (a); i--)
#define YN(b) cout << ((b) ? "YES" : "NO") << "\n";
#define Yn(b) cout << ((b) ? "Yes" : "No") << "\n";
#define yn(b) cout << ((b) ? "yes" : "no") << "\n";

using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vi = vector<int>;
using vvi = vector<vi>;

void solve()
{
    ll m;
    cin >> m;
    string s;
    cin >> s;

    vm cnt(10), sum(10);
    rep(d, 0, 10)
    {
        cnt[d] = (m - d + 9) / 10;
        sum[d] = cnt[d] * (cnt[d] - 1) * 5 + cnt[d] * d;
    }

    int n = s.size();
    vm pr(n + 1, 1);
    rep(i, 0, n)
        pr[i + 1] = pr[i] * cnt[s[i] - '0'];

    mint ans = 0, prod = 1;
    rrep(i, 0, n)
    {
        int c = s[i] - '0';
        ans += pr[i] * prod * sum[c];
        prod *= cnt[c];
        prod *= m;
    }
    cout << ans << "\n";
}

int main()
{
    int t = 1;
    // cin >> t;
    while (t--)
        solve();
}
0