結果

問題 No.1301 Strange Graph Shortest Path
ユーザー neterukunneterukun
提出日時 2020-11-27 22:17:41
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,554 bytes
コンパイル時間 1,032 ms
コンパイル使用メモリ 82,464 KB
実行使用メモリ 123,276 KB
最終ジャッジ日時 2024-07-26 19:40:10
合計ジャッジ時間 18,922 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
52,736 KB
testcase_01 AC 38 ms
52,992 KB
testcase_02 WA -
testcase_03 AC 427 ms
107,008 KB
testcase_04 AC 581 ms
122,572 KB
testcase_05 AC 338 ms
105,856 KB
testcase_06 AC 509 ms
116,828 KB
testcase_07 AC 497 ms
112,532 KB
testcase_08 AC 431 ms
106,112 KB
testcase_09 AC 509 ms
115,328 KB
testcase_10 WA -
testcase_11 AC 546 ms
116,652 KB
testcase_12 AC 524 ms
118,068 KB
testcase_13 AC 484 ms
111,160 KB
testcase_14 AC 511 ms
114,304 KB
testcase_15 AC 502 ms
113,328 KB
testcase_16 AC 574 ms
122,976 KB
testcase_17 AC 513 ms
115,072 KB
testcase_18 AC 461 ms
110,336 KB
testcase_19 AC 525 ms
117,584 KB
testcase_20 AC 509 ms
118,912 KB
testcase_21 AC 490 ms
113,152 KB
testcase_22 AC 517 ms
120,020 KB
testcase_23 AC 517 ms
113,920 KB
testcase_24 AC 521 ms
119,256 KB
testcase_25 AC 539 ms
119,616 KB
testcase_26 AC 513 ms
114,824 KB
testcase_27 AC 513 ms
116,124 KB
testcase_28 AC 484 ms
107,392 KB
testcase_29 WA -
testcase_30 AC 581 ms
118,788 KB
testcase_31 AC 554 ms
120,484 KB
testcase_32 WA -
testcase_33 WA -
testcase_34 AC 513 ms
121,856 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import heapq
import sys
input = sys.stdin.buffer.readline


def dijkstra(start: int, graph: list) -> list:
    """dijkstra法: 始点startから各頂点への最短距離を求める
    計算量: O((E+V)logV)
    """
    INF = 10 ** 18
    n = len(graph)
    dist = [INF] * n
    dist[start] = 0
    q = [(0, start)] # q = [(startからの距離, 現在地)]
    while q:
        d, v = heapq.heappop(q)
        if dist[v] < d:
            continue
        for nxt_v, cost in graph[v]:
            if dist[v] + cost < dist[nxt_v]:
                dist[nxt_v] = dist[v] + cost
                heapq.heappush(q, (dist[nxt_v], nxt_v))
    return dist


n, m = map(int, input().split())
edges = [list(map(int, input().split())) for _ in range(m)]
ans = 0


graph = [[] for i in range(n)]
for u, v, cost, _ in edges:
    u -= 1
    v -= 1
    graph[u].append((v, cost))
    graph[v].append((u, cost))

start = 0
dist = dijkstra(start, graph)
ans += dist[-1]

path = [n - 1]
v = n - 1
while v != 0:
    for prv_v, cost in graph[v]:
        if dist[prv_v] + cost == dist[v]:
            path.append(prv_v)
            v = prv_v
            break

mapping = set([])
for u, v in zip(path, path[1:]):
    mapping.add((u, v))
    mapping.add((v, u))


graph = [[] for i in range(n)]
for u, v, c, d in edges:
    u -= 1
    v -= 1
    if (u, v) in mapping:
        graph[u].append((v, d))
        graph[v].append((u, d))
    else:
        graph[u].append((v, c))
        graph[v].append((u, c))

start = 0
dist = dijkstra(start, graph)
ans += dist[-1]

print(ans)
0