結果
問題 | No.2805 Go to School |
ユーザー | rlangevin |
提出日時 | 2024-07-12 21:27:48 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,096 ms / 2,000 ms |
コード長 | 1,389 bytes |
コンパイル時間 | 167 ms |
コンパイル使用メモリ | 82,276 KB |
実行使用メモリ | 124,496 KB |
最終ジャッジ日時 | 2024-07-16 01:37:42 |
合計ジャッジ時間 | 17,977 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 35 |
ソースコード
from heapq import heappush, heappop inf = float('inf') def dijkstra(s, g, G): # ゴールがない場合はg=-1とする。 N = len(G) def cost(v, m): return v * N + m dist = [inf] * N mindist = [inf] * N seen = [False] * N Q = [cost(0, s)] while Q: c, m = divmod(heappop(Q), N) if seen[m]: continue seen[m] = True dist[m] = c if m == g: return dist #------heapをアップデートする。-------- for u, C in G[m]: if seen[u]: continue newdist = dist[m] + C #------------------------------------ if newdist >= mindist[u]: continue mindist[u] = newdist heappush(Q, cost(newdist, u)) return dist N, M, L, S, E = map(int, input().split()) G = [[] for i in range(N)] for i in range(M): u, v, c = map(int, input().split()) u, v = u - 1, v - 1 G[u].append((v, c)) G[v].append((u, c)) T = list(map(int, input().split())) D0 = dijkstra(0, -1, G) DN = dijkstra(N - 1, -1, G) if D0[-1] <= S + E - 1: print(max(S + 1, D0[-1] + 1)) exit() inf = 10 ** 18 ans = inf for i in range(L): if D0[T[i]-1] >= S + E: continue ans = min(ans, max(S + 1, D0[T[i]-1] + 1) + DN[T[i]-1]) print(ans) if ans < inf//2 else print(-1)