結果
| 問題 | No.2867 NOT FOUND 404 Again |
| コンテスト | |
| ユーザー |
norioc
|
| 提出日時 | 2026-01-06 00:17:09 |
| 言語 | PyPy3 (7.3.17) |
| 結果 |
TLE
|
| 実行時間 | - |
| コード長 | 1,656 bytes |
| 記録 | |
| コンパイル時間 | 505 ms |
| コンパイル使用メモリ | 82,272 KB |
| 実行使用メモリ | 102,916 KB |
| 最終ジャッジ日時 | 2026-01-06 00:17:16 |
| 合計ジャッジ時間 | 6,410 ms |
|
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 2 |
| other | AC * 1 TLE * 1 -- * 16 |
ソースコード
from collections.abc import Iterable
def accum_dp(xs: Iterable, f, op, e, init: dict, *, is_reset=True):
dp = init.copy()
for x in xs:
pp = {} if is_reset else dp.copy()
dp, pp = pp, dp
for fm_key, fm_val in pp.items():
for to_key, to_val in f(fm_key, fm_val, x):
dp[to_key] = op(dp.get(to_key, e), to_val)
return dp
#
# 状態
# 0 = 不一致
# 1 = 4 まで一致
# 2 = 40 まで一致
def f(k, v, x):
lz, st, lt = k # (leading zero, 状態, 未満か)
if lz:
for d in range(1, 10):
nst = 1 if d == 4 else 0
yield (False, nst, True), 1
yield (True, 0, True), 1 # leading zero 維持
else:
for d in range(10):
if not lt and d > x: break
if st == 2 and d == 4: continue # 404
nst = 0
match (st, d):
case (0, 4):
nst = 1
case (_, 4):
nst = 1
case (1, 0):
nst = 2
nlt = lt | (d < x)
yield (False, nst, nlt), v
def op(a, b):
return (a + b) % MOD
def digit_dp() -> int:
digits = [int(c) for c in N]
init = {(True, 0, True): 1}
for d in range(1, digits[0]+1):
lz = False
st = 1 if d == 4 else 0
lt = d < digits[0]
init[lz, st, lt] = init.get((lz, st, lt), 0) + 1
dp = accum_dp(digits[1:], f, op, 0, init)
res = 0
for (lz, _, _), v in dp.items():
if lz: continue
res += v
res %= MOD
return res
MOD = 998244353
N = input()
ans = digit_dp()
print(ans)
norioc