結果
問題 | No.1112 冥界の音楽 |
ユーザー |
![]() |
提出日時 | 2021-05-11 02:34:18 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 78 ms / 2,000 ms |
コード長 | 773 bytes |
コンパイル時間 | 225 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 75,860 KB |
最終ジャッジ日時 | 2024-09-21 08:37:18 |
合計ジャッジ時間 | 3,634 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
ソースコード
def matmul(A,B): # A,B: 行列 res = [[0]*len(B[0]) for _ in [None]*len(A)] for i, resi in enumerate(res): for k, aik in enumerate(A[i]): for j,bkj in enumerate(B[k]): resi[j] += aik*bkj resi[j] %= MOD return res def matpow(A,p): #A^p mod M if p%2: return matmul(A, matpow(A,p-1)) elif p > 0: b = matpow(A,p//2) return matmul(b,b) else: return [[int(i==j) for j in range(len(A))] for i in range(len(A))] MOD = 10**9+7 S = 36 k,m,n = map(int,input().split()) A = [[0]*S for _ in range(S)] for _ in range(m): p,q,r = map(int,input().split()) A[(p-1)*6+q-1][(q-1)*6+r-1] += 1 v = [0]*S v[:6] = [1]*6 v = matmul([v],matpow(A,n-2)) print(sum(v[0][0:S:6])%MOD)