結果

問題 No.1112 冥界の音楽
ユーザー qumazaki
提出日時 2020-07-11 02:08:17
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 128 ms / 2,000 ms
コード長 940 bytes
コンパイル時間 264 ms
コンパイル使用メモリ 82,116 KB
実行使用メモリ 76,812 KB
最終ジャッジ日時 2024-10-11 23:56:43
合計ジャッジ時間 5,874 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines

k,m,n = map(int,readline().split())
pqr = list(map(int,read().split()))
mod = 10**9+7

mat = [[0] * 36 for _ in range(36)]
it = iter(pqr)
for p,q,r in zip(it,it,it):
    p,q,r = p-1,q-1,r-1
    mat[(p*6+q)][(q*6+r)] = 1

def dot_mod(a,b):
    res = [[0] * 36 for _ in range(36)]
    for i in range(36):
        for j in range(36):
            for k in range(36):
                res[i][j] += a[i][k] * b[k][j]
                res[i][j] %= mod
    return res

mat_ex = [mat]
for i in range(65):
    mat_ex.append( dot_mod(mat_ex[-1],mat_ex[-1]) )

mat_n = [[0] * 36 for _ in range(36)]
for i in range(36):
    mat_n[i][i] = 1

x = n-2
for i in range(65):
    if((x>>i)&1):
        mat_n = dot_mod(mat_n,mat_ex[i])

ans = 0
for i in range(6):
    for j in range(6):
        ans += mat_n[i][6*j]
        ans %= mod

print(ans)
0