結果
問題 | No.807 umg tours |
ユーザー |
![]() |
提出日時 | 2023-04-15 11:22:39 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,387 ms / 4,000 ms |
コード長 | 1,857 bytes |
コンパイル時間 | 215 ms |
コンパイル使用メモリ | 81,968 KB |
実行使用メモリ | 168,588 KB |
最終ジャッジ日時 | 2024-10-10 23:36:14 |
合計ジャッジ時間 | 16,477 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 26 |
ソースコード
# 頂点倍化 # 同じ辺のダミー世界を作る # 今の世界からダミー世界には、辺コスト0で辺に従って移動できる、ただし有向辺 # 有向辺であるところが重要、つまり今の世界には戻ってこれない、一度だけ使える # 入力例1のツアー5。片道は今の世界での1から5の最短距離ダイクストラ # 帰り道は1から5の最短距離ダイクストラと1からダミー5への最短距離ダイクストラの小さい方 # つまり頂点からダイクストラを1回行えば十分 N, M = map(int, input().split()) edges = [[] for i in range(N*2+1)] for i in range(M): a, b, c = map(int, input().split()) # 今の世界での無向辺 edges[a].append((b, c)) edges[b].append((a, c)) # ミラーの世界での無向辺 edges[N+a].append((N+b, c)) edges[N+b].append((N+a, c)) # 今の世界からミラーの世界への有向辺、コスト0 edges[a].append((N+b, 0)) edges[b].append((N+a, 0)) from heapq import heappush, heappop INF = 10 ** 18 def dijkstra(s, n, connect): #(始点, ノード数) distance = [INF] * n que = [(0, s)] #(distance, node) distance[s] = 0 confirmed = [False] * n # ノードが確定済みかどうか while que: w,v = heappop(que) if distance[v]<w: continue confirmed[v] = True for to, cost in connect[v]: # ノード v に隣接しているノードに対して if confirmed[to] == False and distance[v] + cost < distance[to]: distance[to] = distance[v] + cost heappush(que, (distance[to], to)) return distance distance = dijkstra(1, N*2+1, edges) for i in range(1, N+1): dist1 = distance[i] dist2 = min(distance[i], distance[i+N]) ans = dist1+dist2 print(ans)