結果

問題 No.2867 NOT FOUND 404 Again
コンテスト
ユーザー norioc
提出日時 2026-01-06 00:17:09
言語 PyPy3
(7.3.17)
結果
TLE  
実行時間 -
コード長 1,656 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 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
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

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)
0