結果

問題 No.2711 Connecting Lights
コンテスト
ユーザー Theta
提出日時 2025-05-26 17:33:16
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 978 ms / 5,000 ms
コード長 774 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 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
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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()
0