結果
問題 |
No.2711 Connecting Lights
|
ユーザー |
|
提出日時 | 2024-04-10 16:41:05 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 887 bytes |
コンパイル時間 | 421 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 154,092 KB |
最終ジャッジ日時 | 2024-10-02 19:07:12 |
合計ジャッジ時間 | 40,792 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 25 WA * 2 |
ソースコード
def main(): N, M, K = map(int, input().split()) MOD = 998244353 dp_table = [[[0 for _ in range(N + 1)] for _ in range(2**N)] for _ in range(M)] for column in range(M): for c_bit in range(2**N): if column == 0: dp_table[column][c_bit][c_bit.bit_count()] += 1 continue for prev_bit in range(2**N): for prev_ctr in range(N + 1): dp_table[column][c_bit][min(prev_ctr, (c_bit & prev_bit).bit_count( ))] += (dp_table[column - 1][prev_bit][prev_ctr]) dp_table[column][c_bit][min( prev_ctr, (c_bit & prev_bit).bit_count())] %= MOD ans = 0 for elm in dp_table[-1]: for v in elm[K:]: ans += v ans %= MOD print(ans) if __name__ == "__main__": main()