結果

問題 No.3628 Sum of Superfibonacci Numbers
コンテスト
ユーザー 👑 loop0919
提出日時 2026-07-10 00:30:51
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 168 ms / 2,000 ms
+ 919µs
コード長 1,775 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 233 ms
コンパイル使用メモリ 95,852 KB
実行使用メモリ 85,632 KB
最終ジャッジ日時 2026-08-14 20:51:56
合計ジャッジ時間 3,877 ms
ジャッジサーバーID
(参考情報)
judge3_0 / judge2_0
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

MOD = 998244353
inv_2 = pow(2, -1, MOD)


def is_good(n):
    digits = [int(d) for d in str(n)]
    return any(digits[i] > digits[i + 1] + digits[i + 2] for i in range(len(digits) - 2))


dist = [0] * 10000
max_good = 10000

for i in range(9999, -1, -1):
    if is_good(i):
        max_good = i
    dist[i] = max_good - i


def count_bad_numbers(X):
    digits = [int(x) for x in str(X)]

    dp = [[0] * 100 for _ in range(2)]
    dp[1][0] = 1

    for x in digits:
        ndp = [[0] * 100 for _ in range(2)]

        for tight in range(2):
            lim = x if tight else 9

            for prev in range(100):
                cur = dp[tight][prev]
                if cur == 0:
                    continue

                a = prev // 10
                b = prev % 10

                for v in range(lim + 1):
                    if a > b + v:
                        continue

                    ntight = tight and (v == lim)
                    to = (10 * prev + v) % 100
                    ndp[ntight][to] += cur
                    ndp[ntight][to] %= MOD

        dp = ndp

    return [(dp[0][val] + dp[1][val]) % MOD for val in range(100)]


def solve():
    N = int(input())

    div = N // 100
    count = count_bad_numbers(div)

    ans = 0

    for suf in range(100):
        for pref in range(100):
            val = 100 * pref + suf
            ans += count[pref] * dist[val]
            ans %= MOD

    ans -= dist[0]
    ans += (N % MOD) * ((N + 1) % MOD) % MOD * inv_2 % MOD

    max_good = 100 * (div + 1)
    for i in range(100 * (div + 1) - 1, N, -1):
        if is_good(i):
            max_good = i
        ans -= max_good - i

    ans %= MOD

    print(ans)


if __name__ == "__main__":
    T = int(input())

    for _ in range(T):
        solve()
0