結果

問題 No.1112 冥界の音楽
コンテスト
ユーザー qumazaki
提出日時 2020-07-11 02:04:57
言語 Python3
(3.14.2 + numpy 2.4.0 + scipy 1.16.3)
結果
RE  
実行時間 -
コード長 966 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 189 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 44,408 KB
最終ジャッジ日時 2024-10-11 23:48:10
合計ジャッジ時間 21,444 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample RE * 3
other RE * 34
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines
import numpy as np

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

mat = np.zeros((36,36), dtype=np.int)
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 = np.zeros((36,36), dtype=np.int)
    for i in range(36):
        for j in range(36):
            for k in range(36):
                res[i][j] += (a[i,k] * b[k,j])%mod
                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 = np.zeros((36,36), dtype=np.int)
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