結果

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

ソースコード

diff # 

import sys
from heapq import * 

input = sys.stdin.readline

INF = 1001001001
mask32 = (1 << 32) - 1

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 が持つのは、(コスト << 32) + 頂点
# dijkstra 1回目
pq = [0]
dp = [INF] * N
dp[0] = 0
parent = [-1] * N
while pq:
    t = heappop(pq)
    cost = t >> 32
    v = t & mask32
    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] << 32) + 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 が持つのは、(コスト << 32) + 頂点
# dijkstra 2回目
pq = [N - 1]
dp = [INF] * N
dp[N - 1] = 0
# parent = [-1] * N
while pq:
    t = heappop(pq)
    cost = t >> 32
    v = t & mask32
    if cost > dp[v]:
        continue
    for nv, dist in enumerate(adj[v]):
        if dp[nv] > dp[v] + dist:
            dp[nv] = dp[v] + dist
            # parent[nv] = v
            heappush(pq, (dp[nv] << 32) + nv)

second_cost = dp[0]

print(first_cost + second_cost)
0