結果
問題 | No.848 なかよし旅行 |
ユーザー | roaris |
提出日時 | 2019-07-26 16:34:22 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,506 bytes |
コンパイル時間 | 277 ms |
コンパイル使用メモリ | 82,020 KB |
実行使用メモリ | 246,628 KB |
最終ジャッジ日時 | 2024-10-07 02:51:21 |
合計ジャッジ時間 | 6,948 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | TLE | - |
testcase_01 | -- | - |
testcase_02 | -- | - |
testcase_03 | -- | - |
testcase_04 | -- | - |
testcase_05 | -- | - |
testcase_06 | -- | - |
testcase_07 | -- | - |
testcase_08 | -- | - |
testcase_09 | -- | - |
testcase_10 | -- | - |
testcase_11 | -- | - |
testcase_12 | -- | - |
testcase_13 | -- | - |
testcase_14 | -- | - |
testcase_15 | -- | - |
testcase_16 | -- | - |
testcase_17 | -- | - |
testcase_18 | -- | - |
testcase_19 | -- | - |
testcase_20 | -- | - |
testcase_21 | -- | - |
testcase_22 | -- | - |
testcase_23 | -- | - |
testcase_24 | -- | - |
testcase_25 | -- | - |
testcase_26 | -- | - |
testcase_27 | -- | - |
testcase_28 | -- | - |
testcase_29 | -- | - |
ソースコード
from heapq import heappush, heappop from collections import defaultdict def dijkstra(start): distance = [10 ** 18 for _ in range(N)] distance[start] = 0 priority_queue = [] for i in range(N): heappush(priority_queue, (distance[i], i)) while len(priority_queue) > 0: current_dist, current_node = heappop(priority_queue) if distance[current_node] < current_dist: continue for next_node, w in adj_dict[current_node]: if distance[next_node] > distance[current_node] + w: distance[next_node] = distance[current_node] + w heappush(priority_queue, (distance[next_node], next_node)) return distance N, M, P, Q, T = map(int, input().split()) adj_dict = defaultdict(list) for _ in range(M): a_i, b_i, c_i = map(int, input().split()) adj_dict[a_i-1].append((b_i-1, c_i)) adj_dict[b_i-1].append((a_i-1, c_i)) distance = [dijkstra(i) for i in range(N)] ans = -10 ** 18 if distance[0][P-1] + distance[P-1][Q-1] + distance[Q-1][0] <= T: print(T) exit() for m1 in range(N): for m2 in range(N): if distance[0][m1] + max(distance[m1][P-1]+distance[P-1][m2], distance[m1][Q-1]+distance[Q-1][m2]) + distance[m2][0] <= T: ans = max(ans, distance[0][m1] + distance[m2][0] + T - (distance[0][m1] + max(distance[m1][P-1]+distance[P-1][m2], distance[m1][Q-1]+distance[Q-1][m2]) + distance[m2][0])) if ans == -10 ** 18: print(-1) else: print(ans)