結果
問題 |
No.2711 Connecting Lights
|
ユーザー |
|
提出日時 | 2025-05-26 17:33:16 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 978 ms / 5,000 ms |
コード長 | 774 bytes |
コンパイル時間 | 583 ms |
コンパイル使用メモリ | 82,404 KB |
実行使用メモリ | 76,600 KB |
最終ジャッジ日時 | 2025-05-26 17:33:26 |
合計ジャッジ時間 | 9,798 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 27 |
ソースコード
from itertools import product def main(): N, M, K = map(int, input().split()) MOD = 998244353 if M == 1: print(pow(2, N, MOD)) return dp_table = [[0 for _ in range(1 << N)] for _ in range(2)] dp_table[0] = [1 for _ in range(1 << N)] for _ in range(M - 1): for prev_light, cur_light in product(range(1 << N), repeat=2): if (prev_light & cur_light).bit_count() < K: continue dp_table[1][cur_light] += dp_table[0][prev_light] dp_table[1][cur_light] %= MOD dp_table[0] = dp_table[1] dp_table[1] = [0 for _ in range(1 << N)] ans = 0 for elm in dp_table[0]: ans += elm ans %= MOD print(ans) if __name__ == "__main__": main()