結果
問題 |
No.788 トラックの移動
|
ユーザー |
|
提出日時 | 2024-06-20 22:15:05 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,035 bytes |
コンパイル時間 | 352 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 128,696 KB |
最終ジャッジ日時 | 2024-06-20 22:15:13 |
合計ジャッジ時間 | 7,557 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | -- * 3 |
other | TLE * 1 -- * 13 |
ソースコード
import sys from heapq import heappop, heappush input = sys.stdin.buffer.readline INF = 10**9 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 = list(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 = 0 for x in range(N): s += T[x] * 2 * dist_all[x][target] m = 0 for x in range(N): if T[x] == 0: continue m = max(m, dist_all[target][x] - dist_all[L][x]) ans = min(ans, s - m) print(ans)