結果
問題 |
No.1112 冥界の音楽
|
ユーザー |
|
提出日時 | 2020-07-11 00:30:23 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 93 ms / 2,000 ms |
コード長 | 1,012 bytes |
コンパイル時間 | 319 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 75,904 KB |
最終ジャッジ日時 | 2024-10-11 20:58:35 |
合計ジャッジ時間 | 3,323 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
ソースコード
#!/usr/bin/env python3 import sys input=sys.stdin.readline mod=10**9+7 def calc(mat1,mat2): n=len(mat1) ret=[[0]*n for _ in range(n)] for i in range(n): for j in range(n): tmp=0 for k in range(n): tmp+=mat1[i][k]*mat2[k][j] tmp%=mod ret[i][j]=tmp return ret k,m,n=map(int,input().split()) n-=2 edge=[list(map(int,input().split())) for _ in range(m)] vec=[0]*(k**2) mat=[[0]*(k**2) for _ in range(k**2)] for p,q,r in edge: if p==1: vec[(p-1)*k+(q-1)]=1 mat[(q-1)*k+(r-1)][(p-1)*k+(q-1)]=1 tmp=[[0]*(k**2) for _ in range(k**2)] for i in range(k**2): tmp[i][i]=1 for i in range(60): if n&(1<<i): tmp=calc(tmp,mat) mat=calc(mat,mat) cnt=[0]*(k**2) for i in range(k**2): tcnt=0 for j in range(k**2): tcnt+=tmp[i][j]*vec[j] tcnt%=mod cnt[i]=tcnt ans=0 for i in range(k): ans+=cnt[i*k] ans%=mod print(ans)