結果

問題 No.1301 Strange Graph Shortest Path
コンテスト
ユーザー zkou
提出日時 2020-11-04 23:14:30
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,198 bytes
コンパイル時間 419 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 75,540 KB
最終ジャッジ日時 2024-09-13 00:38:03
合計ジャッジ時間 36,536 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 28 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
from heapq import * 

input = sys.stdin.readline

INF = float("inf")

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 が持つのは、(コスト, 頂点)
# dijkstra 1回目
pq = [(0, 0)]
dp = [INF] * N
dp[0] = 0
parent = [-1] * N
while pq:
    cost, v = heappop(pq)
    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], nv))
# print(dp)
v = N - 1
while v != 0:
    p = parent[v]
    adj[v][p] = adj[p][v] = uv2d[(v, p)]
    v = p

# pq が持つのは、(コスト, 頂点)
# dijkstra 2回目
pq = [(dp[N - 1], N - 1)]
for i in range(N - 1):
    dp[i] = INF
# parent = [-1] * N
while pq:
    cost, v = heappop(pq)
    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], nv))
# print(dp)
print(dp[0])
0