結果

問題 No.1112 冥界の音楽
ユーザー tikitaka1201
提出日時 2020-07-12 15:05:19
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,222 ms / 2,000 ms
コード長 890 bytes
コンパイル時間 150 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 11,008 KB
最終ジャッジ日時 2024-10-15 03:38:59
合計ジャッジ時間 5,505 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff #

def mat_mul(A, B, MOD=None):
    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]
                if MOD!=None:
                    C[i][j] %= MOD
    return C

def mat_pow(A, n, MOD=None):
    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, MOD)
        n >>= 1
        A = mat_mul(A, A, MOD)
    return res

k,m,n = map(int, input().split())
mod = 10**9+7
A = [[0]*(k**2) for _ in range(k**2)]

for _ in range(m):
    p,q,r = map(int, input().split())
    s = (p-1)*k+q-1
    t = (q-1)*k+r-1
    A[s][t] = 1

res = mat_pow(A, n-2, mod)
ans = 0

for i in range(k):
    for j in range(k):
        ans+=res[i][j*k]
        ans%=mod

print(ans)
0