結果

問題 No.2867 NOT FOUND 404 Again
コンテスト
ユーザー PNJ
提出日時 2024-09-03 02:55:37
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,523 bytes
コンパイル時間 260 ms
コンパイル使用メモリ 82,664 KB
実行使用メモリ 182,804 KB
最終ジャッジ日時 2024-09-03 02:55:47
合計ジャッジ時間 10,369 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 3 WA * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 998244353
N = input()
le = len(N)
dp = [[0,0,0,0] for n in range(le)]
for n in range(le):
  if n == 0:
    for i in range(int(N[n])):
      if i == 4:
        dp[n][1] = 1
      else:
        dp[n][0] += 1
    p,q,r,s = 0,0,0,0
    if N[n] == '4':
      q += 1
    else:
      p += 1
  else:
    for i in range(10):
      if i == 0:
        dp[n][0] = (dp[n][0] + dp[n - 1][0] + dp[n - 1][2]) % mod
        dp[n][2] = (dp[n][2] + dp[n - 1][1]) % mod
        dp[n][3] = (dp[n][3] + dp[n - 1][3]) % mod
      elif i == 4:
        dp[n][1] = (dp[n][1] + dp[n - 1][0] + dp[n - 1][1]) % mod
        dp[n][3] = (dp[n][3] + dp[n - 1][2] + dp[n - 1][3]) % mod
      else:
        dp[n][0] = (dp[n][0] + dp[n - 1][0] + dp[n - 1][1] + dp[n - 1][2]) % mod
        dp[n][3] = (dp[n][3] + dp[n - 1][3]) % mod
    for i in range(int(N[n])):
      if p == 1:
        if i == 4:
          dp[n][1] += 1
        else:
          dp[n][0] += 1
      elif q == 1:
        if i == 0:
          dp[n][2] += 1
        else:
          dp[n][0] += 1
      elif r == 1:
        if i  == 4:
          dp[n][3] += 1
        else:
          dp[n][0] += 1
      elif s == 1:
        dp[n][3] += 1
    
    if s == 1:
      continue

    if N[n] == '4':
      if r == 1:
        s = 1
        r = 0
      else:
        p = 0
        q = 1
    elif N[n] == '0':
      if q == 1:
        r = 1
        q = 0
      else:
        p,q,r = 1,0,0
    else:
      p,q,r = 1,0,0
ans = dp[-1][0] + dp[-1][1] + dp[-1][2] + p + q + r - 1
ans %= mod
print(ans)
0