結果

問題 No.1750 ラムドスウイルスの感染拡大-hard
ユーザー flippergo
提出日時 2025-04-13 11:21:36
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,161 ms / 2,000 ms
コード長 963 bytes
コンパイル時間 651 ms
コンパイル使用メモリ 82,276 KB
実行使用メモリ 80,836 KB
最終ジャッジ日時 2025-04-13 11:21:52
合計ジャッジ時間 15,101 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353
N,M,T = map(int,input().split())
G = {i:[] for i in range(N)}
for _ in range(M):
    s,t = map(int,input().split())
    G[s].append(t)
    G[t].append(s)
A = [[0 for _ in range(N)] for _ in range(N)]
for i in range(N):
    for j in G[i]:
        A[i][j] = 1
def matmul(A,B):
    C = [[0 for _ in range(N)] for _ in range(N)]
    for i in range(N):
        for j in range(N):
            for k in range(N):
                C[i][j] += A[i][k]*B[k][j]
                C[i][j] %= MOD
    return C
def matpow(A,k):
    if k==0:
        return [[1 if i==j else 0 for j in range(N)] for i in range(N)]
    if k==1:
        return A
    if k%2==0:
        B = matpow(A,k//2)
        return matmul(B,B)
    else:
        B = matpow(A,k//2)
        return matmul(matmul(B,B),A)
B = matpow(A,T)
u = [0 for _ in range(N)]
u[0] = 1
v = [0 for _ in range(N)]
for i in range(N):
    for j in range(N):
        v[i] += B[i][j]*u[j]
        v[i] %= MOD
print(v[0])
0