#717827 (PyPy3) No.1750 ラムドスウイルスの感染拡大-hard

提出ソース

結果

問題	No.1750 ラムドスウイルスの感染拡大-hard
ユーザー	👑 SPD_9X2
提出日時	2021-11-19 21:51:16
言語	PyPy3 (7.3.15)
結果	TLE (最新) AC (最初)
実行時間	-
コード長	1,292 bytes
コンパイル時間	497 ms
コンパイル使用メモリ	87,028 KB
実行使用メモリ	85,168 KB
最終ジャッジ日時	2023-08-30 08:11:06
合計ジャッジ時間	23,250 ms
ジャッジサーバーID （参考情報）	judge15 / judge12

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	70 ms 71,304 KB
testcase_01	AC	69 ms 71,396 KB
testcase_02	AC	74 ms 75,296 KB
testcase_03	AC	75 ms 76,292 KB
testcase_04	AC	112 ms 76,992 KB
testcase_05	AC	71 ms 71,376 KB
testcase_06	AC	70 ms 71,512 KB
testcase_07	AC	70 ms 71,276 KB
testcase_08	AC	291 ms 78,100 KB
testcase_09	AC	266 ms 78,124 KB
testcase_10	AC	303 ms 78,076 KB
testcase_11	AC	295 ms 78,028 KB
testcase_12	AC	394 ms 78,260 KB
testcase_13	AC	387 ms 78,272 KB
testcase_14	TLE	-
testcase_15	TLE	-
testcase_16	TLE	-
testcase_17	AC	1,698 ms 84,676 KB
testcase_18	AC	1,576 ms 84,704 KB
testcase_19	AC	1,632 ms 84,428 KB
testcase_20	AC	1,334 ms 82,728 KB
testcase_21	AC	1,757 ms 83,876 KB
testcase_22	AC	287 ms 78,268 KB
testcase_23	TLE	-
testcase_24	AC	216 ms 77,496 KB
testcase_25	AC	474 ms 80,372 KB
testcase_26	AC	175 ms 77,180 KB
testcase_27	AC	81 ms 76,184 KB
testcase_28	AC	89 ms 76,900 KB
testcase_29	AC	81 ms 76,192 KB
testcase_30	AC	276 ms 78,220 KB
testcase_31	AC	260 ms 78,120 KB
testcase_32	AC	259 ms 78,100 KB
testcase_33	AC	246 ms 77,700 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

"""

行列べき乗

"""

import sys
from sys import stdin

#行列A,Bの積(不可能だとエラー,modに0以下の値を指定するとmodを取らなくなる)
def matrix_mul(A,B,mod):
 
    ans = [ [0] * len(B[0]) for i in range(len(A)) ]
 
    for ai in range(len(A)):
 
        for bj in range(len(B[0])):
 
            now = 0
            for same in range(len(A[0])):
                now += A[ai][same] * B[same][bj]

            if mod > 0:
                ans[ai][bj] = now % mod
            else:
                ans[ai][bj] = now
 
    return ans
 
#行列Aのx乗(当然正方行列じゃないとだめ)
def matrix_pow(A,x,mod):
 
    B = [[A[i][j] for j in range(len(A[0]))] for i in range(len(A))]
    ans = [[0] * len(A[0]) for i in range(len(A))]
    for i in range(len(A)):
        ans[i][i] = 1
 
    while x > 0:
        if x % 2 == 1:
            ans = matrix_mul(ans,B,mod)
        B = matrix_mul(B,B,mod)
 
        x//=2
    return ans


N,M,T = map(int,stdin.readline().split())

A = [ [0] * N ]
A[0][0] = 1

H = [ [0] * N for i in range(N) ]

for i in range(M):

    u,v = map(int,stdin.readline().split())
    #u -= 1
    #v -= 1

    H[u][v] += 1
    H[v][u] += 1

mod = 998244353
ans = matrix_mul(A , matrix_pow(H,T,mod) , mod )
print (ans[0][0])
#print (ans)

yukicoder

結果

テストケース

ソースコード