結果
問題 | 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 sysinput = sys.stdin.readlinefrom 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] %= MODreturn Cdef mat_pow(A, n):res = [[0]*len(A) for _ in range(len(A))]for i in range(len(A)):res[i][i] = 1while n>0:if n&1:res = mat_mul(res, A)n >>= 1A = mat_mul(A, A)return resN, 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] = 1A[v][u] = 1x = [[0] for _ in range(N)]x[0][0] = 1MOD = 998244353res = mat_mul(mat_pow(A, T), x)print(res[0][0])