結果

問題 No.2867 NOT FOUND 404 Again
コンテスト
ユーザー rlangevin
提出日時 2024-11-17 12:43:15
言語 C++17(clang)
(17.0.6 + boost 1.87.0)
結果
AC  
実行時間 2,660 ms / 3,000 ms
コード長 1,843 bytes
コンパイル時間 1,516 ms
コンパイル使用メモリ 122,240 KB
実行使用メモリ 8,272 KB
最終ジャッジ日時 2024-11-17 12:43:58
合計ジャッジ時間 40,514 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <string>

using namespace std;

const int MOD = 998244353;

int main() {
    string input;
    cin >> input;

    vector<int> N(input.size());
    for (size_t i = 0; i < input.size(); ++i) {
        N[i] = input[i] - '0';
    }

    vector<int> pre0(100, 0), pre1(100, 0);

    if (N.size() <= 2) {
        string ans = "";
        for (int n : N) {
            ans += to_string(n);
        }
        cout << ans << endl;
        return 0;
    }

    int st = N[0] * 10 + N[1];
    for (int i = 0; i < st; ++i) {
        pre0[i] = 1;
    }
    pre1[st] = 1;
    vector<int> dp0(100, 0), dp1(100, 0);
    for (size_t k = 2; k < N.size(); ++k) {
        int n = N[k];
        fill(dp0.begin(), dp0.end(), 0);
        fill(dp1.begin(), dp1.end(), 0);
        for (int a = 0; a < 10; ++a) {
            for (int b = 0; b < 10; ++b) {
                if (pre0[a*10+b] == 0 && pre1[a*10+b] == 0) continue;
                for (int c = 0; c < 10; ++c) {
                    if (a == 4 && b == 0 && c == 4) continue;
                    dp0[b * 10 + c] = (dp0[b * 10 + c] + pre0[a * 10 + b]) % MOD;
                }
                if (a != 4 || b != 0 || n != 4) {
                    dp1[b * 10 + n] = (dp1[b * 10 + n] + pre1[a * 10 + b]) % MOD;
                }
                for (int c = 0; c < n; ++c) {
                    if (a == 4 && b == 0 && c == 4) continue;
                    dp0[b * 10 + c] = (dp0[b * 10 + c] + pre1[a * 10 + b]) % MOD;
                }
            }
        }
        pre0.swap(dp0);
        pre1.swap(dp1);
    }

    int ans = -1;
    for (int i = 0; i < 10; ++i) {
        for (int j = 0; j < 10; ++j) {
            ans = (ans + pre0[i * 10 + j]) % MOD;
            ans = (ans + pre1[i * 10 + j]) % MOD;
        }
    }

    cout << ans << endl;
    return 0;
}
0