結果

問題 No.1750 ラムドスウイルスの感染拡大-hard
ユーザー stng
提出日時 2022-05-15 13:50:17
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,002 bytes
コンパイル時間 483 ms
コンパイル使用メモリ 82,096 KB
実行使用メモリ 76,968 KB
最終ジャッジ日時 2024-07-26 20:40:43
合計ジャッジ時間 26,138 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 23 TLE * 7
権限があれば一括ダウンロードができます

ソースコード

diff # 

def dot(A1,A2,n,mod):
  # n:行列のサイズ
  A = [[0]*n for _ in range(n)]
  for i in range(n):
    for j in range(n):
      for k in range(n):
        A[i][j] += A1[i][k]*A2[k][j]
        A[i][j] %= mod
  return A

def pow_mat(A,k,n,mod):
  # A:累乗する行列, k:累乗数, n:行列Aのサイズ
  P = [[0]*n for _ in range(n)]
  for i in range(n): P[i][i] = 1
  while k:
    if k&1: P = dot(P,A,n,mod)
    A = dot(A,A,n,mod)
    k >>= 1
  return P

n,m,T = map(int,input().split())
st = [[] for i in range(n)]
mod = 998244353

st = [[0]*n for i in range(n)]

for i in range(m):
    s,t = map(int,input().split())
    st[s][t] = 1
    st[t][s] = 1

"""
ct = [0]*n
ct[0] = 1

for i in range(T):
    nct = [0]*n
    for j in range(n):
        for k in st[j]:
            nct[j] += ct[k]
            nct[j] %= mod
    ct = nct[:]

print(ct[0])
"""

A = pow_mat(st,T,n,mod)
#ans = 0
"""
for i in range(n):
    for j in range(n):
        ans += A[i][j]
        ans %= mod
"""        
print(A[0][0])
0