結果
問題 |
No.2711 Connecting Lights
|
ユーザー |
|
提出日時 | 2024-04-07 15:48:23 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
WA
|
実行時間 | - |
コード長 | 717 bytes |
コンパイル時間 | 150 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 44,348 KB |
最終ジャッジ日時 | 2024-10-01 04:34:16 |
合計ジャッジ時間 | 16,930 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 20 WA * 7 |
ソースコード
import typing import sys input = lambda: sys.stdin.readline().strip() inf = 10**18 mod = 998244353 import numpy as np 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 dp = np.ones(1<<N, dtype=int) Ans = np.identity(1<<N, dtype=int) i = 0 M -= 1 while M: if M % 2 == 1: Ans = Ans@As[i] Ans%=mod M //= 2 i += 1 Ans = dp@Ans ans = sum(Ans)%mod print(ans) def main(): t = 1 for _ in range(t): solve() main()