結果
問題 |
No.788 トラックの移動
|
ユーザー |
|
提出日時 | 2024-06-20 22:19:59 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 970 bytes |
コンパイル時間 | 449 ms |
コンパイル使用メモリ | 82,348 KB |
実行使用メモリ | 113,300 KB |
最終ジャッジ日時 | 2024-06-20 22:20:14 |
合計ジャッジ時間 | 14,579 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 8 WA * 2 TLE * 4 |
ソースコード
import sys from heapq import heappop, heappush input = sys.stdin.buffer.readline INF = 10**18 def dijkstra(N, G, s): dist = [INF] * N que = [(0, s)] dist[s] = 0 while que: c, v = heappop(que) if dist[v] < c: continue for t, cost in G[v]: if dist[v] + cost < dist[t]: dist[t] = dist[v] + cost heappush(que, (dist[t], t)) return dist N, M, L = map(int, input().split()) L -= 1 T = tuple(map(int, input().split())) G = [[] for _ in range(N)] for _ in range(M): a, b, c = map(int, input().split()) a -= 1 b -= 1 G[a].append((b, c)) G[b].append((a, c)) dist_all = [] for source in range(N): dist_all.append(dijkstra(N, G, source)) ans = INF for target in range(N): s = sum(T[x] * 2 * dist_all[x][target] for x in range(N)) m = max(dist_all[target][x] - dist_all[L][x] for x in range(N) if T[x] != 0) ans = min(ans, s - m) print(ans)