結果

問題 No.1340 おーじ君をさがせ
コンテスト
ユーザー ygd.
提出日時 2021-01-17 00:32:23
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 858 bytes
コンパイル時間 493 ms
コンパイル使用メモリ 82,192 KB
実行使用メモリ 81,140 KB
最終ジャッジ日時 2024-11-28 15:42:24
合計ジャッジ時間 27,272 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 39 WA * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

def mat_dot(A, B, MOD=pow(10,9)+7):
  #A:n*m行列, B:m*l行列
  n, m, l = len(A), len(A[0]), len(B[0])
  X = [[0]*l for i in range(n)]
  for i in range(n):
    for j in range(l):
      temp = 0
      for k in range(m):
        temp = (temp|(A[i][k]&B[k][j])) % MOD
      X[i][j] = temp
  return X
 
def mat_pow(A, x, MOD=pow(10,9)+7):
  #A^x
  n = len(A)
  X = [[0] * n for i in range(n)]
  for i in range(n):
    X[i][i] = 1
  for i in range(x.bit_length()):
    if (x>>i)&1:
      X = mat_dot(X, A, MOD)
    A = mat_dot(A, A, MOD)
  return X

N,M,T = map(int,input().split())
A = [[0]*N for _ in range(N)] 
for i in range(M):
    a,b = map(int,input().split())
    A[a][b] = 1

AT = mat_pow(A,T)
I = [[0] for _ in range(N)]
#print(I)
I[0][0] = 1
#print(AT)
ANS = mat_dot(AT,I)
#print(ANS)
ret = 0
for x in ANS:
    for y in x:
        ret += y
print(ret)
0