結果

問題 No.1301 Strange Graph Shortest Path
コンテスト
ユーザー zkou
提出日時 2020-11-04 22:45:54
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,250 bytes
コンパイル時間 448 ms
コンパイル使用メモリ 82,180 KB
実行使用メモリ 133,396 KB
最終ジャッジ日時 2024-09-13 00:36:21
合計ジャッジ時間 20,024 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 28 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
from heapq import * 

input = sys.stdin.readline

INF = 10 ** 15

N, M = map(int, input().split())
adj = [dict() for _ in range(N)]
uv2d = dict()
for _ in range(M):
    u, v, c, d = map(int, input().split())
    u -= 1
    v -= 1
    adj[u][v] = adj[v][u] = c
    uv2d[(u, v)] = uv2d[(v, u)] = d

# pq が持つのは、(コスト * N) + 頂点
# dijkstra 1回目
pq = [0]
dp = [INF] * N
dp[0] = 0
parent = [-1] * N
while pq:
    cost, v = divmod(heappop(pq), N)
    if cost > dp[v]:
        continue
    for nv, dist in adj[v].items():
        if dp[nv] > dp[v] + dist:
            dp[nv] = dp[v] + dist
            parent[nv] = v
            heappush(pq, (dp[nv] * N) + nv)

first_cost = dp[N - 1]

v = N - 1
while v != 0:
    p = parent[v]
    adj[v][p] = adj[p][v] = uv2d[(v, p)]
    v = p

# pq が持つのは、(コスト * N) + 頂点
# dijkstra 2回目
pq = [N - 1]
dp = [INF] * N
dp[N - 1] = 0
# parent = [-1] * N
while pq:
    cost, v = divmod(heappop(pq), N)
    if cost > dp[v]:
        continue
    for nv, dist in adj[v].items():
        if dp[nv] > dp[v] + dist:
            dp[nv] = dp[v] + dist
            # parent[nv] = v
            heappush(pq, (dp[nv] * N) + nv)

second_cost = dp[0]

print(first_cost + second_cost)
0