結果

問題 No.2711 Connecting Lights
コンテスト
ユーザー Theta
提出日時 2024-04-10 16:44:02
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 4,051 ms / 5,000 ms
コード長 948 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 175 ms
コンパイル使用メモリ 85,120 KB
実行使用メモリ 158,720 KB
最終ジャッジ日時 2026-04-19 02:36:43
合計ジャッジ時間 30,919 ms
ジャッジサーバーID
（参考情報） 		judge1_1 / judge2_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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