結果
問題 | No.2569 はじめてのおつかいHard |
ユーザー | loop0919 |
提出日時 | 2023-11-14 00:43:40 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 735 ms / 2,000 ms |
コード長 | 1,166 bytes |
コンパイル時間 | 296 ms |
コンパイル使用メモリ | 81,700 KB |
実行使用メモリ | 120,468 KB |
最終ジャッジ日時 | 2023-11-14 00:43:47 |
合計ジャッジ時間 | 6,383 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 261 ms
91,128 KB |
testcase_01 | AC | 253 ms
91,128 KB |
testcase_02 | AC | 262 ms
91,128 KB |
testcase_03 | AC | 256 ms
91,128 KB |
testcase_04 | AC | 263 ms
91,128 KB |
testcase_05 | AC | 726 ms
120,468 KB |
testcase_06 | AC | 735 ms
105,192 KB |
testcase_07 | AC | 478 ms
119,480 KB |
testcase_08 | AC | 650 ms
120,364 KB |
testcase_09 | AC | 486 ms
100,632 KB |
testcase_10 | AC | 37 ms
53,460 KB |
testcase_11 | AC | 38 ms
53,460 KB |
ソースコード
from heapq import heappush, heappop INF = 1 << 60 # ダイクストラ法 def dijkstra(start, G): costs = [INF]*len(G) costs[start] = 0 pq = [] heappush(pq, (0, start)) while pq: now_t, now_v = heappop(pq) if costs[now_v] < now_t: continue for next_t, next_v in G[now_v]: if costs[next_v] > costs[now_v] + next_t: costs[next_v] = costs[now_v] + next_t heappush(pq, (now_t + next_t, next_v)) return costs N, M = map(int, input().split()) graph = [[] for _ in range(N)] graph_rev = [[] for _ in range(N)] for _ in range(M): u, v, t = map(int, input().split()) u -= 1; v -= 1 graph[u].append((t, v)) graph_rev[v].append((t, u)) di = { N-2: dijkstra(N-2, graph), N-1: dijkstra(N-1, graph) } di_rev = { N-2: dijkstra(N-2, graph_rev), N-1: dijkstra(N-1, graph_rev) } for i in range(N-2): ans_1 = di_rev[N-2][i] + di[N-2][N-1] + di[N-1][i] ans_2 = di_rev[N-1][i] + di[N-1][N-2] + di[N-2][i] ans = min(ans_1, ans_2) if ans < INF: print(ans) else: print(-1)