結果
問題 |
No.1112 冥界の音楽
|
ユーザー |
![]() |
提出日時 | 2020-07-10 22:59:52 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 172 ms / 2,000 ms |
コード長 | 1,126 bytes |
コンパイル時間 | 558 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 74,240 KB |
最終ジャッジ日時 | 2024-10-11 13:41:05 |
合計ジャッジ時間 | 4,140 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
ソースコード
def _mul(A, B, MOD): C = [[0] * len(B[0]) for i in range(len(A))] for i in range(len(A)): for k in range(len(B)): for j in range(len(B[0])): C[i][j] = (C[i][j] + A[i][k] * B[k][j]) % MOD return C def pow_matrix(A, n, MOD): """A**nをダブリングによって求める。""" B = [[0] * 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 = _mul(A, B, MOD) A = _mul(A, A, MOD) n = n // 2 return B k, m, n = map(int, input().split()) info = [list(map(int, input().split())) for i in range(m)] MOD = 10 ** 9 + 7 # matrix_l * a * matrix_r matrix = [[0] * (k ** 2) for i in range(k ** 2)] for p, q, r in info: p -= 1 q -= 1 r -= 1 row = 0 col = 0 row += r * k row += q col += q * k col += p matrix[row][col] += 1 matrix = pow_matrix(matrix, n - 2, MOD) ans = [0] * (k ** 2) for i in range(k): ans[i * k] = 1 res = 0 for i in range(k): for j in range(k ** 2): res += matrix[i][j] * ans[j] res %= MOD print(res)