結果

問題 No.2867 NOT FOUND 404 Again
コンテスト
ユーザー flippergo
提出日時 2025-03-10 17:54:10
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,915 ms / 3,000 ms
コード長 1,651 bytes
コンパイル時間 388 ms
コンパイル使用メモリ 82,180 KB
実行使用メモリ 423,148 KB
最終ジャッジ日時 2025-03-10 17:54:47
合計ジャッジ時間 31,584 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353
A = list(input().strip())
N = len(A)
A = ["0"]+A
A = [int(A[i]) for i in range(N+1)]
dp = [[[0 for _ in range(2)] for _ in range(3)] for _ in range(N+1)]
if A[1]>4:
    dp[1][0][0] = 1
    dp[1][1][1] = 1
    dp[1][0][1] = A[1]-1
elif A[1]==4:
    dp[1][1][0] = 1
    dp[1][0][1] = 4
else:
    dp[1][0][0] = 1
    dp[1][0][1] = A[1]
for i in range(1,N):
    dp[i+1][0][1] = (dp[i+1][0][1]+9*dp[i][0][1]+8*dp[i][1][1]+9*dp[i][2][1])%MOD
    dp[i+1][1][1] = (dp[i+1][1][1]+dp[i][0][1]+dp[i][1][1])%MOD
    dp[i+1][2][1] = (dp[i+1][2][1]+dp[i][1][1])%MOD
    if A[i+1]>4:
        dp[i+1][0][0] = (dp[i+1][0][0]+dp[i][0][0]+dp[i][1][0]+dp[i][2][0])%MOD
        dp[i+1][1][1] = (dp[i+1][1][1]+dp[i][0][0]+dp[i][1][0])%MOD
        dp[i+1][0][1] = (dp[i+1][0][1]+(A[i+1]-1)*dp[i][0][0]+(A[i+1]-2)*dp[i][1][0]+(A[i+1]-1)*dp[i][2][0])%MOD
        dp[i+1][2][1] = (dp[i+1][2][1]+dp[i][1][0])%MOD
    elif A[i+1]==4:
        dp[i+1][1][0] = (dp[i+1][1][0]+dp[i][0][0]+dp[i][1][0])%MOD
        dp[i+1][0][1] = (dp[i+1][0][1]+A[i+1]*dp[i][0][0]+A[i+1]*dp[i][2][0]+(A[i+1]-1)*dp[i][1][0])%MOD
        dp[i+1][2][1] = (dp[i+1][2][1]+dp[i][1][0])%MOD
    elif 0<A[i+1]<4:
        dp[i+1][0][0] = (dp[i+1][0][0]+dp[i][0][0]+dp[i][1][0]+dp[i][2][0])%MOD
        dp[i+1][0][1] = (dp[i+1][0][1]+A[i+1]*dp[i][0][0]+(A[i+1]-1)*dp[i][1][0]+A[i+1]*dp[i][2][0])%MOD
        dp[i+1][2][1] = (dp[i+1][2][1]+dp[i][1][0])%MOD
    else:
        dp[i+1][0][0] = (dp[i+1][0][0]+dp[i][0][0]+dp[i][2][0])%MOD
        dp[i+1][2][0] = (dp[i+1][2][0]+dp[i][1][0])%MOD
ans = 0
for j in range(3):
    for s in range(2):
        ans = (ans+dp[N][j][s])%MOD
print((ans-1)%MOD)
0