結果
問題 | No.1340 おーじ君をさがせ |
ユーザー |
![]() |
提出日時 | 2021-01-15 23:32:16 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 948 bytes |
コンパイル時間 | 195 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 80,640 KB |
最終ジャッジ日時 | 2024-11-26 20:19:53 |
合計ジャッジ時間 | 15,691 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 43 WA * 16 |
ソースコード
import sys input = sys.stdin.buffer.readline # nxnの行列積 # n is necessary n, m, t = map(int, input().split()) graph = [[0] * (n) for _ in range(n)] for _ in range(m): a, b = map(int, input().split()) a, b = a - 1, b - 1 graph[a][b] = 1 def matrix_product(A, B): res = [[0] * n for _ in range(n)] for i in range(n): for k in range(n): for j in range(n): res[i][j] += A[i][k] and B[k][j] for i in range(n): for j in range(n): if res[i][j]: res[i][j] = 1 return res def matrix_exponents(A, t): res = [[0] * n for _ in range(n)] for i in range(n): res[i][i] = 1 while t > 0: if t & 1 == 1: res = matrix_product(res, A) A = matrix_product(A, A) t //= 2 return res res = matrix_exponents(graph, t - 1) ans = 0 for j in range(n): if res[0][j]: ans += 1 print(ans)