結果

問題 No.3242 Count 8 Included Numbers (Hard)
ユーザー Theta
提出日時 2025-08-26 13:51:23
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,517 ms / 2,000 ms
コード長 2,491 bytes
コンパイル時間 331 ms
コンパイル使用メモリ 82,888 KB
実行使用メモリ 302,436 KB
最終ジャッジ日時 2025-09-04 18:06:10
合計ジャッジ時間 25,361 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 5
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #

def main():
    MOD = 998244353
    N = input()

    dp_table = [[[0, 0] for _ in range(2)] for _ in range(len(N) + 1)]
    dp_table[0][1][0] = 1

    for idx, cur_n_str in enumerate(N):
        cur_n = int(cur_n_str)
        for next_n in range(10):
            # 確実に上限に当たらない
            if next_n < cur_n:
                if next_n == 8:
                    dp_table[idx + 1][0][1] += dp_table[idx][0][0] + \
                        dp_table[idx][0][1] + \
                        dp_table[idx][1][0] + dp_table[idx][1][1]
                    dp_table[idx + 1][0][1] %= MOD
                else:
                    dp_table[idx + 1][0][0] += dp_table[idx][0][0] + \
                        dp_table[idx][1][0]
                    dp_table[idx + 1][0][1] += dp_table[idx][0][1] + \
                        dp_table[idx][1][1]
                    dp_table[idx + 1][0][0] %= MOD
                    dp_table[idx + 1][0][1] %= MOD
            elif next_n == cur_n:
                # ちょうど上限
                if next_n == 8:
                    dp_table[idx + 1][0][1] += dp_table[idx][0][0] + \
                        dp_table[idx][0][1]
                    dp_table[idx + 1][1][1] += dp_table[idx][1][0] + \
                        dp_table[idx][1][1]
                    dp_table[idx + 1][0][1] %= MOD
                    dp_table[idx + 1][1][1] %= MOD
                else:
                    dp_table[idx + 1][0][0] += dp_table[idx][0][0]
                    dp_table[idx + 1][0][1] += dp_table[idx][0][1]
                    dp_table[idx + 1][1][0] += dp_table[idx][1][0]
                    dp_table[idx + 1][1][1] += dp_table[idx][1][1]
                    dp_table[idx + 1][0][0] %= MOD
                    dp_table[idx + 1][0][1] %= MOD
                    dp_table[idx + 1][1][0] %= MOD
                    dp_table[idx + 1][1][1] %= MOD
            else:
                # 上限を超える
                if next_n == 8:
                    dp_table[idx + 1][0][1] += dp_table[idx][0][0] + \
                        dp_table[idx][0][1]
                    dp_table[idx + 1][0][1] %= MOD
                else:
                    dp_table[idx + 1][0][0] += dp_table[idx][0][0]
                    dp_table[idx + 1][0][1] += dp_table[idx][0][1]
                    dp_table[idx + 1][0][0] %= MOD
                    dp_table[idx + 1][0][1] %= MOD
    print((dp_table[-1][0][1] + dp_table[-1][1][1]) % MOD)


if __name__ == "__main__":
    main()
0