結果

問題 No.1112 冥界の音楽
コンテスト
ユーザー mkawa2
提出日時 2020-07-10 22:33:33
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,467 bytes
コンパイル時間 265 ms
コンパイル使用メモリ 82,200 KB
実行使用メモリ 92,756 KB
最終ジャッジ日時 2024-10-11 09:55:01
合計ジャッジ時間 14,732 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 31 WA * 2 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

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)]

def dot(aa,bb):
    return [[sum(a*b for a,b in zip(row,col))%md for col in zip(*bb)] for row in aa]

def solve(n):
    if n == 3:
        ans = 0
        for p, q, r in pqr:
            if p == 1 and q == 1: ans += 1
        print(ans)
        exit()

    start = []
    end = []
    to = [[0] * m for _ in range(m)]
    for i in range(m):
        if pqr[i][0] == 1: start.append(i)
        if pqr[i][2] == 1: end.append(i)
        for j in range(m):
            if pqr[i][1:] == pqr[j][:2]: to[i][j] = 1
    # print(start,end)
    # print(to)

    cnt = [[0] * i + [1] + [0] * (m - i - 1) for i in range(m)]
    n -= 3
    while n:
        if n & 1: cnt = dot(cnt, to)
        to = dot(to, to)
        n >>= 1
    # print(cnt)

    ans = 0
    for s in start:
        for e in end:
            ans += cnt[s][e]
            ans %= md
    print(ans)

md=10**9+7
k,m,n=MI()
pqr=LLI(m)
solve(n)
0