結果
問題 | No.788 トラックの移動 |
ユーザー |
![]() |
提出日時 | 2021-09-16 14:08:15 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,085 bytes |
コンパイル時間 | 164 ms |
コンパイル使用メモリ | 82,032 KB |
実行使用メモリ | 88,064 KB |
最終ジャッジ日時 | 2024-06-29 14:23:17 |
合計ジャッジ時間 | 9,980 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 12 WA * 2 |
ソースコード
import heapq INF = 10**18 def dijkstra(s, edge): n = len(edge) dist = [INF]*n dist[s] = 0 edgelist = [] heapq.heappush(edgelist,(dist[s], s)) while edgelist: minedge = heapq.heappop(edgelist) if dist[minedge[1]] < minedge[0]: continue v = minedge[1] for e in edge[v]: if dist[e[1]] > dist[v]+e[0]: dist[e[1]] = dist[v]+e[0] heapq.heappush(edgelist,(dist[e[1]], e[1])) return dist import sys import io, os input = io.BytesIO(os.read(0,os.fstat(0).st_size)).readline n, m, l = map(int, input().split()) l -= 1 T = list(map(int, input().split())) g = [[] for i in range(n)] for i in range(m): a, b, c = map(int, input().split()) a, b = a-1, b-1 g[a].append((c, b)) g[b].append((c, a)) d1 = dijkstra(l, g) ans = INF for i in range(n): d2 = dijkstra(i, g) s = 0 for j in range(n): s += d2[j]*T[j] for j in range(n): if T[j] == 0: continue temp = 2*s+d1[j]-d2[j] ans = min(ans, temp) print(ans)