結果
問題 | No.1887 K Consecutive Ks (Easy) |
ユーザー |
![]() |
提出日時 | 2022-03-25 23:50:06 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 440 ms / 2,000 ms |
コード長 | 629 bytes |
コンパイル時間 | 244 ms |
コンパイル使用メモリ | 82,784 KB |
実行使用メモリ | 206,868 KB |
最終ジャッジ日時 | 2024-10-14 08:07:13 |
合計ジャッジ時間 | 4,453 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 16 |
ソースコード
MOD = 998244353n, m = map(int, input().split())dp = [[0] * (m + 1) for _ in range(n + 1)] # dp[i][k] = # of length i array ending j meeting the conditionsno = [[0] * (m + 1) for _ in range(n + 1)] # no[i][j] = # of length i array ending j not meeting the conditionsno[0][0] = 1for i in range(n):no_sum = sum(no[i]) % MODfor k in range(1, min(m + 1, n - i + 1)):dp[i + k][k] = (dp[i + k][k] + no_sum - no[i][k]) % MODfor k in range(2, m + 1):no[i + 1][k] = (no_sum - dp[i + 1][k]) % MODans = 0for i in range(1, n + 1):ans += sum(dp[i]) * pow(m, n - i, MOD)ans %= MODprint(ans)