結果

問題 No.1964 sum = length
コンテスト
ユーザー tktk_snsn
提出日時 2022-06-03 23:10:00
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 922 bytes
コンパイル時間 299 ms
コンパイル使用メモリ 82,436 KB
実行使用メモリ 78,880 KB
最終ジャッジ日時 2024-09-21 03:15:29
合計ジャッジ時間 11,546 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 19 WA * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict
mod = 998244353
U = 211


def id(x, y):
    return x * U + y


def parse(x):
    a = x // U
    return a, x - U * a


def dec(x):
    if x < 10:
        return 1
    if x < 100:
        return 2
    return 3


"""
sum + 1 == len

1: sum += 1, len += 2
2: sum += 2, len += 2
x: sum += x, len += dec(x) + 1
"""


n = int(input())
dp = defaultdict(int)
dp[0] = 1
for i in range(n):
    ok = n - i - 1 + 10
    ndp = defaultdict(int)
    for k, v in dp.items():
        s, l = parse(k)
        for a in range(1, U):
            ns = s + a
            nl = l + dec(a) + 1
            if ns >= U or nl >= U:
                break
            if ns - nl > ok:
                break
            nk = id(ns, nl)
            ndp[nk] += v
            ndp[nk] %= mod
    dp = ndp

ans = 0
for k, v in dp.items():
    s, l = parse(k)
    if s + 1 == l:
        ans += v
        ans %= mod
print(ans)
0