結果
問題 |
No.1112 冥界の音楽
|
ユーザー |
![]() |
提出日時 | 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 |
ソースコード
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)