結果

問題 No.1750 ラムドスウイルスの感染拡大-hard
ユーザー roaris
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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])
0