結果

問題 No.1750 ラムドスウイルスの感染拡大-hard
ユーザー horitaka1999horitaka1999
提出日時 2021-11-19 23:15:00
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,081 ms / 2,000 ms
コード長 742 bytes
コンパイル時間 274 ms
コンパイル使用メモリ 82,308 KB
実行使用メモリ 85,124 KB
最終ジャッジ日時 2025-01-01 21:57:26
合計ジャッジ時間 14,062 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

N,M,T = map(int,input().split())
MOD = 998244353
def mulmod(a,b,mod):
    return a*b%mod
def mat_mul_Pypy(A,B):
    return [[sum([mulmod(A[i][k],B[k][j],MOD) for k in range(len(B))])%MOD for j in range(len(B[0]))] for i in range(len(A))]

def mat_power_Pypy(A, N):#行列累乗
    P = [[1 if i==j else 0 for i in range(len(A[0]))] for j in range(len(A))]
    while N:
        if N & 1:
            P = mat_mul_Pypy(P, A)
        A = mat_mul_Pypy(A, A)
        N >>= 1
    return P
mat = [[0] * N for _ in range(N)]
for _ in range(M):
    s,t = map(int,input().split())
    mat[s][t] = 1
    mat[t][s] = 1
P = mat_power_Pypy(mat,T)
initial = [[0] * 1 for _ in range(N)]
initial[0][0] = 1
ans_mat = mat_mul_Pypy(P,initial)
print(ans_mat[0][0])
0