結果
問題 |
No.848 なかよし旅行
|
ユーザー |
|
提出日時 | 2022-02-06 22:00:56 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 537 ms / 2,000 ms |
コード長 | 1,099 bytes |
コンパイル時間 | 339 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 103,812 KB |
最終ジャッジ日時 | 2024-06-11 14:11:13 |
合計ジャッジ時間 | 7,460 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 26 |
ソースコード
from heapq import heappop, heappush INF = 1 << 60 def dijkstra(s, edge): n = len(edge) dist = [INF] * n dist[s] = 0 hq = [] heappush(hq, (dist[s], s)) while hq: dist_v, v = heappop(hq) if dist_v > dist[v]: continue for cost, nx in edge[v]: if dist[nx] > dist[v] + cost: dist[nx] = dist[v] + cost heappush(hq, (dist[nx], nx)) return dist N, M, P, Q, T = map(int, input().split()) P -= 1 Q -= 1 G = [[] for _ in range(N)] for _ in range(M): a, b, c = map(int, input().split()) a -= 1 b -= 1 G[a].append((c, b)) G[b].append((c, a)) dist_0 = dijkstra(0, G) dist_p = dijkstra(P, G) dist_q = dijkstra(Q, G) tmp = dist_0[P] + dist_p[Q] + dist_q[0] if tmp <= T: print(T) exit() ans = -1 for u in range(N): for v in range(N): tmp = dist_0[u] tmp += max(dist_p[u] + dist_p[v], dist_q[u] + dist_q[v]) tmp += dist_0[v] if tmp <= T: ans = max(ans, T - max(dist_p[u] + dist_p[v], dist_q[u] + dist_q[v])) print(ans)