結果

問題 No.1301 Strange Graph Shortest Path
コンテスト
ユーザー nehan_der_thal
提出日時 2020-11-27 22:07:40
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,295 bytes
コンパイル時間 407 ms
コンパイル使用メモリ 82,184 KB
実行使用メモリ 161,384 KB
最終ジャッジ日時 2024-09-13 00:58:50
合計ジャッジ時間 26,725 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 28 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

from heapq import heappush, heappop
INF = 10**18
def dijkstra(N, G, s):
    dist = [INF] * N
    parents = [-1]*N
    que = [(0, s)]
    dist[s] = 0
    while que:
        c, v = heappop(que)
        if dist[v] < c:
            continue
        for t, cost in G[v]:
            if dist[v] + cost < dist[t]:
                dist[t] = dist[v] + cost
                parents[t] = v
                heappush(que, (dist[t], t))
    return dist, parents

N, M = map(int, input().split())
G = [set() for _ in range(N)]
vs = list()
X = []
for _ in range(M):
    a, b, c, d = map(int, input().split())
    a-=1;b-=1
    G[a].add((b, c))
    G[b].add((a, c))
    X.append((a, b, c, d))
dd, par = dijkstra(N, G, 0)
ans = dd[N-1]

#c = N-1
#R = [N-1]
#while 1:
#    mn = 10**18
#    rr = -1
#    for u, co in G[c]:
#        if dd[u]+co < mn:
#            mn = co+dd[u]
#            rr = u
#    R.append(rr)
#    c = rr
#    if rr == 0:
#        break

c = N-1
R = [N-1]
while c!=0:
    c = par[c]
    R.append(c)
    
G2 = [set() for _ in range(N)]
vs = set()
for u, v in zip(R, R[1:]):
    vs.add((u, v))
    vs.add((v, u))
for a, b, c, d in X:
    if (a, b) in vs:
        co = d
    else:
        co = c
    G2[a].add((b, co))
    G2[b].add((a, co))

dd, _ = dijkstra(N, G2, N-1)
ans += dd[0]
print(ans)
0