結果

問題 No.2867 NOT FOUND 404 Again
コンテスト
ユーザー tottoripaper
提出日時 2024-09-03 02:16:17
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 317 ms / 3,000 ms
コード長 1,305 bytes
コンパイル時間 2,082 ms
コンパイル使用メモリ 198,172 KB
最終ジャッジ日時 2025-02-24 03:46:03
ジャッジサーバーID
（参考情報） 		judge2 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

int main(){
    std::cin.tie(nullptr);
    std::ios::sync_with_stdio(false);

    std::string S;
    std::cin >> S;

    int n = S.size();
    std::vector dp(n + 1, std::vector<int>(6, 0));
    dp[0][1 + 0 * 2] = 1;

    constexpr int MOD = 998244353;
    auto trans = [](int ai, int ni, int s){
        int s1 = s % 2, s2 = s / 2;
        int ns1;
        if(s1 == 1){
            ns1 = ai <= ni;
        }else{
            ns1 = ai < ni;
        }
        int ns2;
        if(s2 == 0 && ai == 4){
            ns2 = 1;
        }else if(s2 == 1 && ai == 0){
            ns2 = 2;
        }else if(s2 == 2 && ai == 4){
            ns2 = 3;
        }else if(ai == 4){
            ns2 = 1;
        }else{
            ns2 = 0;
        }
        return ns1 + ns2 * 2;
    };
    for(int i=0;i<n;i++){
        for(int s=0;s<6;s++){
            for(int ai=0;ai<=9;ai++){
                int ns = trans(ai, S[n - 1 - i] - '0', s);
                if(ns / 2 < 3){
                    dp[i + 1][ns] += dp[i][s];
                    if(dp[i + 1][ns] >= MOD){dp[i + 1][ns] -= MOD;}
                }
            }
        }
    }

    int res = -1;
    for(int s2=0;s2<3;s2++){
        res += dp[n][1 + s2 * 2];
        if(res >= MOD){res -= MOD;}
    }

    std::cout << res << std::endl;
}
0