結果

問題 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
権限があれば一括ダウンロードができます

ソースコード

diff #

# 頂点倍化
# 同じ辺のダミー世界を作る
# 今の世界からダミー世界には、辺コスト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)
0