結果

問題 No.1301 Strange Graph Shortest Path
ユーザー nehan_der_thal
提出日時 2020-11-27 21:56:13
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,382 bytes
コンパイル時間 349 ms
コンパイル使用メモリ 82,220 KB
実行使用メモリ 160,224 KB
最終ジャッジ日時 2024-09-13 00:55:42
合計ジャッジ時間 26,026 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 28 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

# V: 頂点数
# g[v] = {(w, cost)}:
#     頂点vから遷移可能な頂点(w)とそのコスト(cost)
# r: 始点の頂点

from heapq import heappush, heappop
INF = 10**18
def dijkstra(N, G, s):
    dist = [INF] * 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
                heappush(que, (dist[t], t))
    return dist

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
#    if a==0:
#        R = min(c+d, R)
#        vs.append((b, c, d))
    G[a].add((b, c))
    G[b].add((a, c))
    X.append((a, b, c, d))
dd = dijkstra(N, G, 0)
ans = dd[N-1]

c = N-1
k = 0
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
    
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:
        G2[a].add((b, d))
        G2[b].add((a, d))
    else:
        G2[a].add((b, c))
        G2[b].add((a, c))

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

print(ans)
0