結果

問題 No.1112 冥界の音楽
ユーザー tikitaka1201
提出日時 2020-07-12 15:05:57
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 182 ms / 2,000 ms
コード長 890 bytes
コンパイル時間 347 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 74,112 KB
最終ジャッジ日時 2024-10-15 03:39:46
合計ジャッジ時間 3,723 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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