結果
問題 | No.1750 ラムドスウイルスの感染拡大-hard |
ユーザー |
![]() |
提出日時 | 2021-11-19 22:23:14 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,100 ms / 2,000 ms |
コード長 | 857 bytes |
コンパイル時間 | 263 ms |
コンパイル使用メモリ | 82,380 KB |
実行使用メモリ | 77,420 KB |
最終ジャッジ日時 | 2025-01-01 18:57:47 |
合計ジャッジ時間 | 14,042 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 30 |
ソースコード
import sys input = sys.stdin.readline from collections import * def mat_mul(A, B): C = [[0]*len(B[0]) for _ in range(len(A))] for i in range(len(A)): for j in range(len(B[0])): for k in range(len(A[0])): C[i][j] += A[i][k]*B[k][j] C[i][j] %= MOD return C def mat_pow(A, n): res = [[0]*len(A) for _ in range(len(A))] for i in range(len(A)): res[i][i] = 1 while n>0: if n&1: res = mat_mul(res, A) n >>= 1 A = mat_mul(A, A) return res N, M, T = map(int, input().split()) A = [[0]*N for _ in range(N)] for _ in range(M): u, v = map(int, input().split()) A[u][v] = 1 A[v][u] = 1 x = [[0] for _ in range(N)] x[0][0] = 1 MOD = 998244353 res = mat_mul(mat_pow(A, T), x) print(res[0][0])