結果

問題 No.1112 冥界の音楽
コンテスト
ユーザー mkawa2
提出日時 2020-07-10 22:36:57
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,197 bytes
コンパイル時間 198 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 44,564 KB
最終ジャッジ日時 2024-10-11 09:58:43
合計ジャッジ時間 22,794 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22 WA * 11 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

import numpy as np
import sys

sys.setrecursionlimit(10 ** 6)

def II(): return int(sys.stdin.readline())
def MI(): return map(int, sys.stdin.readline().split())
def LI(): return list(map(int, sys.stdin.readline().split()))
def SI(): return sys.stdin.readline()[:-1]
def LLI(rows_number): return [LI() for _ in range(rows_number)]
def LLI1(rows_number): return [LI1() for _ in range(rows_number)]
int1 = lambda x: int(x) - 1
def MI1(): return map(int1, sys.stdin.readline().split())
def LI1(): return list(map(int1, sys.stdin.readline().split()))
p2D = lambda x: print(*x, sep="\n")
dij = [(1, 0), (0, 1), (-1, 0), (0, -1)]

md=10**9+7
k,m,n=MI()
pqr=LLI(m)
if n==3:
    ans=0
    for p,q,r in pqr:
        if p==1 and r==1:ans+=1
    print(ans)
    exit()
to=np.zeros((m,m),dtype="i8")
for i in range(m):
    for j in range(m):
        if pqr[i][1:]==pqr[j][:2]:to[i][j]=1
#print(to)
cnt=np.eye(m,dtype="i8")
n-=3
while n:
    if n&1:cnt=np.dot(cnt,to)%md
    to=np.dot(to,to)%md
    n>>=1
#print(cnt)

start=[]
end=[]
for i,(p,q,r) in enumerate(pqr):
    if p==1:start.append(i)
    if r==1:end.append(i)
ans=0
for s in start:
    for e in end:
        ans+=cnt[s][e]
        ans%=md
print(ans)
0