結果

問題 No.2772 Appearing Even Times
コンテスト
ユーザー norioc
提出日時 2025-05-21 01:23:37
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 567 ms / 4,000 ms
コード長 1,651 bytes
コンパイル時間 967 ms
コンパイル使用メモリ 83,932 KB
実行使用メモリ 326,144 KB
最終ジャッジ日時 2025-05-21 01:23:46
合計ジャッジ時間 8,585 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <string>
using namespace std;

const long long MOD = 998244353;

long long digit_dp(const string& s) {
    vector<int> ds;
    for (char c : s) ds.push_back(c - '0');
    int nd = ds.size();

    // dp[i][j][k][m]
    // i : i 桁目までみた
    // j : n 未満か(j=0 完全一致 j=1 より小さい)
    // k : leading zero
    // m : 各数字の頻度の偶奇（ビットで表現）
    vector<vector<vector<vector<long long>>>> dp(
        nd + 1,
        vector<vector<vector<long long>>>(
            2,
            vector<vector<long long>>(
                2,
                vector<long long>(1 << 10, 0)
            )
        )
    );

    dp[0][0][1][0] = 1;

    for (int i = 0; i < nd; ++i) {
        for (int j = 0; j < 2; ++j) {
            int to = (j == 0) ? ds[i] : 9;
            for (int k = 0; k < 2; ++k) {
                for (int m = 0; m < (1 << 10); ++m) {
                    long long cur = dp[i][j][k][m];
                    if (cur == 0) continue;
                    for (int x = 0; x <= to; ++x) {
                        int nj = j | (x < to);
                        int nk = k & (x == 0);
                        int nm = nk ? m : (m ^ (1 << x));
                        dp[i + 1][nj][nk][nm] += cur;
                        if (dp[i + 1][nj][nk][nm] >= MOD)
                            dp[i + 1][nj][nk][nm] -= MOD;
                    }
                }
            }
        }
    }

    long long res = (dp[nd][0][0][0] + dp[nd][1][0][0]) % MOD;
    return res;
}

int main() {
    string N;
    cin >> N;
    cout << digit_dp(N) << endl;
    return 0;
}
0