結果
問題 | No.848 なかよし旅行 |
ユーザー |
|
提出日時 | 2021-02-24 18:39:38 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 1,474 ms / 2,000 ms |
コード長 | 708 bytes |
コンパイル時間 | 208 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 223,380 KB |
最終ジャッジ日時 | 2024-11-08 01:27:16 |
合計ジャッジ時間 | 37,876 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
外部呼び出し有り |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 26 |
ソースコード
import sys import numpy as np from scipy.sparse.csgraph import dijkstra from scipy.sparse import csr_matrix input = sys.stdin.buffer.readline N, M, P, Q, T = map(int, input().split()) P -= 1 Q -= 1 edges = np.array([tuple(map(int, input().split())) for _ in range(M)]).T matr = csr_matrix((edges[2], (edges[:2] - 1)), shape=(N, N)) way_1, way_p, way_q = dijkstra(matr, indices=(0, P, Q), directed=False).astype(int) if way_1[P] + way_1[Q] + way_p[Q] <= T: print(T) exit() xpy = way_p[:, None] + way_p[None, :] xqy = way_q[:, None] + way_q[None, :] xy1 = way_1[:, None] + way_1[None, :] res = np.where((xy1 + xpy <= T) & (xy1 + xqy <= T), T - np.maximum(xpy, xqy), -1) ans = res.max() print(ans)