結果

問題 No.1750 ラムドスウイルスの感染拡大-hard
ユーザー Mine
提出日時 2022-01-07 23:01:29
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 667 bytes
コンパイル時間 507 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 23,936 KB
最終ジャッジ日時 2024-11-14 09:06:24
合計ジャッジ時間 67,312 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 8 TLE * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

def mul(a, b, mod=998244353):
    n = len(a)
    ans = [[0]*n for i in range(n)]
    for i in range(n):
        for j in range(n):
            for k in range(n):
                ans[i][j] += a[i][k] * b[k][j]
                ans[i][j] %= mod
    return ans


n, m, t = map(int, input().split())
mod = 998244353
graph = [[0]*n for i in range(n)]
for i in range(m):
    a, b = map(int, input().split())
    graph[a][b] = 1
    graph[b][a] = 1

ans = [[0]*n for i in range(n)]
for i in range(n):
    ans[i][i] = 1

t_bin = bin(t)[2:]
for i in range(len(t_bin))[::-1]:
    if t_bin[i] == "1":
        ans = mul(graph, ans)
    graph = mul(graph, graph)

print(ans[0][0])
0