結果
問題 |
No.848 なかよし旅行
|
ユーザー |
![]() |
提出日時 | 2020-06-19 00:23:59 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 617 ms / 2,000 ms |
コード長 | 1,158 bytes |
コンパイル時間 | 658 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 103,756 KB |
最終ジャッジ日時 | 2024-11-08 01:17:07 |
合計ジャッジ時間 | 7,645 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 26 |
ソースコード
import heapq def dijkstra(s, graph): n = len(graph)-1 dist = [float("inf") for i in range(n+1)] dist[s] = 0 pq = [] heapq.heapify(pq) heapq.heappush(pq, (0, s)) while pq: mini_dis, node = heapq.heappop(pq) if dist[node] < mini_dis: continue for w, point in graph[node]: if dist[point] < w: continue newlen = dist[node]+w if newlen < dist[point]: heapq.heappush(pq, (newlen, point)) dist[point] = newlen return dist N, M, P, Q, T = map(int, input().split()) graph = [[] for i in range(N+1)] for i in range(M): a, b, c = map(int, input().split()) graph[a].append((c, b)) graph[b].append((c, a)) A = dijkstra(1, graph) B = dijkstra(P, graph) C = dijkstra(Q, graph) if max(2*A[P], 2*A[Q]) > T: print(-1) exit() if (A[P]+A[Q]+C[P]) <= T: print(T) exit() else: ans = 0 for x in range(1, N+1): for y in range(1, N+1): if A[x]+B[x]+B[y]+A[y] <= T and A[x]+C[x]+C[y]+A[y] <= T: ans = max(ans,T-max(B[x]+B[y],C[x]+C[y])) print(ans)