結果
問題 |
No.1112 冥界の音楽
|
ユーザー |
|
提出日時 | 2020-07-11 10:27:12 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
RE
(最新)
AC
(最初)
|
実行時間 | - |
コード長 | 599 bytes |
コンパイル時間 | 295 ms |
コンパイル使用メモリ | 12,416 KB |
実行使用メモリ | 44,488 KB |
最終ジャッジ日時 | 2024-10-12 18:42:40 |
合計ジャッジ時間 | 22,864 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | RE * 3 |
other | RE * 34 |
ソースコード
import numpy as np MOD = 10 ** 9 + 7 int1 = lambda x: int(x) - 1 def pow_matrix_mod(x, n, mod=MOD): if not n: return np.eye(len(x), dtype=np.object) if n % 2 == 0: return pow_matrix_mod(x @ x % mod, n // 2) % mod else: return x @ pow_matrix_mod(x @ x % mod, (n - 1) // 2) % mod K, M, N = map(int, input().split()) matr = np.zeros((36, 36), dtype=np.object) for _ in range(M): P, Q, R = map(int1, input().split()) matr[6 * P + Q, 6 * Q + R] = True A = np.zeros(36, dtype=np.object) A[:6] = 1 print((A @ pow_matrix_mod(matr, N - 2))[::6].sum() % MOD)