結果

問題 No.2711 Connecting Lights
コンテスト
ユーザー Theta
提出日時 2024-04-10 16:44:02
言語 PyPy3
(7.3.15)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 948 bytes
コンパイル時間 836 ms
コンパイル使用メモリ 82,400 KB
実行使用メモリ 153,676 KB
最終ジャッジ日時 2024-10-02 19:10:39
合計ジャッジ時間 44,053 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

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