結果
問題 |
No.1112 冥界の音楽
|
ユーザー |
![]() |
提出日時 | 2020-11-29 23:22:43 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 234 ms / 2,000 ms |
コード長 | 1,077 bytes |
コンパイル時間 | 204 ms |
コンパイル使用メモリ | 82,420 KB |
実行使用メモリ | 76,880 KB |
最終ジャッジ日時 | 2024-09-13 02:12:02 |
合計ジャッジ時間 | 3,827 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
ソースコード
import sys def MI(): return map(int,sys.stdin.readline().rstrip().split()) def matrix_multiplication_mod(A,B,p): l = len(A) m = len(B) n = len(B[0]) C = [[0]*n for _ in range(l)] for i in range(l): for j in range(n): C[i][j] = sum((A[i][k]*B[k][j]) % p for k in range(m)) % p return C def matrix_power_mod(A,n,p): l = len(A) C = [[0] * l for _ in range(l)] for i in range(l): C[i][i] = 1 while n > 0: if n % 2 == 1: C = matrix_multiplication_mod(C,A,p) A = matrix_multiplication_mod(A,A,p) n >>= 1 return C K,M,N = MI() mod = 10**9+7 PQR = set([tuple(MI()) for _ in range(M)]) A = [[0]*(K**2) for _ in range(K**2)] for i in range(K**2): i0,i1 = divmod(i,K) for j in range(K**2): j0,j1 = divmod(j,K) if i0 != j1: continue if (j0+1,i0+1,i1+1) in PQR: A[i][j] = 1 X = matrix_power_mod(A,N-2,mod) ans = 0 for i in range(0,K**2,K): for j in range(K): ans += X[i][j] ans %= mod print(ans)