結果
問題 |
No.2711 Connecting Lights
|
ユーザー |
|
提出日時 | 2024-04-07 15:42:28 |
言語 | PyPy3 (7.3.15) |
結果 |
RE
|
実行時間 | - |
コード長 | 843 bytes |
コンパイル時間 | 291 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 69,120 KB |
最終ジャッジ日時 | 2024-10-01 04:33:20 |
合計ジャッジ時間 | 3,339 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | RE * 3 |
other | RE * 27 |
ソースコード
import typing import sys # sys.setrecursionlimit(100100) from collections import defaultdict input = lambda: sys.stdin.readline().strip() inf = 10**18 mod = 998244353 import numpy as np # import heapq def solve(): N, M, K = map(int, input().split()) A = [[0 for _ in range(1 << N)] for _ in range(1 << N)] for i in range(1 << N): for j in range(1 << N): if (i&j).bit_count() >= K: A[i][j] = 1 As = [np.array(A)] for _ in range(15): As.append(As[-1]@As[-1]) As[-1]%=mod i = 0 if M == 1: print(0) return dp = sum(As[0]).reshape((-1, 1)) Ans = np.identity(1<<N, dtype=int) M-=2 while M: if M % 2 == 1: Ans = Ans@As[i] Ans%=mod M //= 2 i += 1 Ans = Ans@dp ans = sum(Ans)[0]%mod print(ans) def main(): t = 1 for _ in range(t): solve() main()