結果

問題 No.2867 NOT FOUND 404 Again
コンテスト
ユーザー Tekyla
提出日時 2024-08-30 22:35:34
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
1/20  
(最新)
WA  
(最初)
実行時間 -
コード長 727 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
最終ジャッジ日時 2024-08-30 22:35:44
ジャッジサーバーID
（参考情報） 		judge5_1 / judge4_0
このコードへのチャレンジ
（要ログイン）

ソースコード

diff #
raw source code 

N = input()
P = 998244353

dp_eq = [1, 0, 0] #0文字一致, 1文字一致, 2文字一致
dp = [0,0,0]
for n in N:
    dp = [(dp[0]*9 + dp[1]*8 + dp[2]*9) % P, (dp[0]+dp[1]) % P, dp[1]]
    for i in range(int(n)):
        if i == 0:
            dp[0] += (dp_eq[0] + dp_eq[2])
            dp[2] += dp_eq[1]
        elif i == 4:
            dp[1] += dp_eq[0] + dp_eq[1]
        else:
            dp[0] += sum(dp_eq)
    if n == "0" and dp_eq[1] > 0:
        dp_eq = [0,0,1]
    elif n == "4":
        if dp_eq[2] > 0:
            dp_eq = [0,0,0]
        else:
            dp_eq = [0, 1, 0]
    else:
        dp_eq = [1,0,0]

if "404" in N:
    print((sum(dp) + sum(dp_eq) - 2) % P)
else:
    print((sum(dp) + sum(dp_eq) - 1) % P)
0