結果

問題 No.807 umg tours
ユーザー AEnAEn
提出日時 2022-05-18 10:18:21
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,499 ms / 4,000 ms
コード長 913 bytes
コンパイル時間 310 ms
コンパイル使用メモリ 81,792 KB
実行使用メモリ 150,520 KB
最終ジャッジ日時 2024-09-16 11:06:46
合計ジャッジ時間 24,133 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 46 ms
61,440 KB
testcase_01 AC 52 ms
61,952 KB
testcase_02 AC 62 ms
64,512 KB
testcase_03 AC 53 ms
63,616 KB
testcase_04 AC 46 ms
61,056 KB
testcase_05 AC 49 ms
61,568 KB
testcase_06 AC 54 ms
64,384 KB
testcase_07 AC 52 ms
63,232 KB
testcase_08 AC 36 ms
52,352 KB
testcase_09 AC 39 ms
52,992 KB
testcase_10 AC 38 ms
53,504 KB
testcase_11 AC 1,067 ms
113,664 KB
testcase_12 AC 1,329 ms
119,424 KB
testcase_13 AC 1,625 ms
128,776 KB
testcase_14 AC 744 ms
102,056 KB
testcase_15 AC 521 ms
94,984 KB
testcase_16 AC 1,557 ms
130,772 KB
testcase_17 AC 2,176 ms
145,384 KB
testcase_18 AC 2,274 ms
145,764 KB
testcase_19 AC 1,964 ms
141,644 KB
testcase_20 AC 1,069 ms
120,836 KB
testcase_21 AC 1,188 ms
123,840 KB
testcase_22 AC 513 ms
95,748 KB
testcase_23 AC 447 ms
92,120 KB
testcase_24 AC 1,111 ms
147,360 KB
testcase_25 AC 2,499 ms
150,520 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from heapq import heappush, heappop
INF = float('inf')

N, M = map(int, input().split())
adj = [[] for _ in range(N)]
for i in range(M):
    s, t, d = map(int, input().split())
    s -= 1
    t -= 1
    adj[s].append((t, d))
    adj[t].append((s, d))

def dijkstra(s, n):
    dist = [[INF] * 2 for _ in range(N)]
    hq = [(0, 0, s)]
    dist[s][0] = 0
    dist[s][1] = 0
    while hq:
        d, ticket, v = heappop(hq)
        if d > dist[v][ticket]:
            continue
        for to, cost in adj[v]:
            if dist[v][ticket] + cost < dist[to][ticket]:
                dist[to][ticket] = d + cost
                # prev[to] = v
                heappush(hq, (dist[to][ticket], ticket, to))
            if ticket == 0:
                if dist[to][1] > d:
                    dist[to][1] = d
                    heappush(hq, (d, 1, to))
    return dist
dis = dijkstra(0, N)
for a, b in dis:
    print(a+b)
0