結果
| 問題 |
No.807 umg tours
|
| コンテスト | |
| ユーザー |
FromBooska
|
| 提出日時 | 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)
FromBooska