結果
問題 |
No.2711 Connecting Lights
|
ユーザー |
|
提出日時 | 2024-04-07 23:52:14 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 98 ms / 5,000 ms |
コード長 | 1,044 bytes |
コンパイル時間 | 179 ms |
コンパイル使用メモリ | 82,396 KB |
実行使用メモリ | 75,204 KB |
最終ジャッジ日時 | 2024-10-01 04:48:48 |
合計ジャッジ時間 | 3,747 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 27 |
ソースコード
import typing import sys input = lambda: sys.stdin.readline().strip() inf = 10**18 mod = 998244353 # numpyのオーバーフローが原因だった # 行列積 def matmul2d(A, B, mod): assert len(A[0]) == len(B) n = len(A) l = len(A[0]) m = len(B[0]) C = [[0 for _ in range(m)] for _ in range(n)] for i in range(n): for j in range(m): for k in range(l): C[i][j] += A[i][k]*B[k][j] C[i][j] %= mod return C 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 = [A] for _ in range(18): As.append(matmul2d(As[-1], As[-1], mod)) Ans = [[1 if i==j else 0 for j in range(1 << N)] for i in range(1 << N)] i = 0 M-=1 while M: if M % 2 == 1: Ans = matmul2d(Ans, As[i], mod) M //= 2 i += 1 print(sum(sum(a) for a in Ans)%mod) def main(): t = 1 for _ in range(t): solve() main()