結果
問題 |
No.3242 Count 8 Included Numbers (Hard)
|
ユーザー |
|
提出日時 | 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 |
ソースコード
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()