結果
問題 |
No.1112 冥界の音楽
|
ユーザー |
|
提出日時 | 2022-05-22 16:33:52 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 115 ms / 2,000 ms |
コード長 | 933 bytes |
コンパイル時間 | 306 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 76,160 KB |
最終ジャッジ日時 | 2024-09-20 12:42:08 |
合計ジャッジ時間 | 3,781 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
ソースコード
# 行列の乗算(mod) def mat_mul(a, b): I, K, J = len(a), len(b), len(b[0]) c = [[0 for j in range(J)] for i in range(I)] for i in range(I): for k in range(K): for j in range(J): c[i][j] += a[i][k] * b[k][j] c[i][j] %= mod return c # 行列の累乗(mod) def mat_pow(a, n): b = [[0 for j in range(len(a))] for i in range(len(a))] for i in range(len(a)): b[i][i] = 1 while n > 0: if n & 1: b = mat_mul(b, a) a = mat_mul(a, a) n >>= 1 return b K,m,n = map(int,input().split()) mod = 10**9+7 dp = [[0]*(K*K) for i in range(K*K)] for i in range(m): p,q,r = map(int,input().split()) p -= 1 q -= 1 r -= 1 pq = p*K+q qr = q*K+r dp[pq][qr] = 1 A = mat_pow(dp,n-2) ans = 0 for i in range(K): for j in range(K): ans += A[i][j*K] ans %= mod print(ans) #print(A)