結果
問題 | No.1750 ラムドスウイルスの感染拡大-hard |
ユーザー |
|
提出日時 | 2021-11-20 16:36:15 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 958 ms / 2,000 ms |
コード長 | 832 bytes |
コンパイル時間 | 277 ms |
コンパイル使用メモリ | 82,532 KB |
実行使用メモリ | 77,300 KB |
最終ジャッジ日時 | 2024-06-11 18:29:37 |
合計ジャッジ時間 | 12,116 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 30 |
ソースコード
mod = 998244353 def mat_mul(a, b): n_a, m_a = len(a), len(a[0]) n_b, m_b = len(b), len(b[0]) res = [[0] * m_b for i in range(n_a)] for i in range(n_a): for k in range(m_a): for j in range(m_b): res[i][j] += a[i][k] * b[k][j] res[i][j] %= mod return res def mat_pow(a, k): if k == 1: return a n = len(a) res = [[0] * n for i in range(n)] for i in range(n): res[i][i] = 1 while k: if k & 1: res = mat_mul(res, a) a = mat_mul(a, a) k >>= 1 return res n, m, t = map(int, input().split()) G = [[0] * n for i in range(n)] for _ in range(m): a, b = map(int, input().split()) G[a][b] = 1 G[b][a] = 1 G = mat_pow(G, t) b = [[0] * n for i in range(n)] b[0][0] = 1 G = mat_mul(G, b) print(G[0][0])