結果
問題 |
No.1995 CHIKA Road
|
ユーザー |
![]() |
提出日時 | 2023-10-17 21:11:14 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 689 ms / 2,000 ms |
コード長 | 1,504 bytes |
コンパイル時間 | 169 ms |
コンパイル使用メモリ | 82,560 KB |
実行使用メモリ | 155,048 KB |
最終ジャッジ日時 | 2024-09-17 18:23:10 |
合計ジャッジ時間 | 13,564 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 37 |
ソースコード
# Nは大きすぎるのでMにある頂点とスタートとゴールだけ結ぶ # それらの頂点も隣とだけ結べばいいだろう # N個が大きすぎてedgesも張れないので必要nodesだけに絞る N, M = map(int, input().split()) nodes = set() nodes.add(1) nodes.add(N) edge_list = [] for i in range(M): a, b = map(int, input().split()) edge_list.append((a, b, 2*b-2*a-1)) nodes.add(a) nodes.add(b) nodes = sorted(list(nodes)) dic = {} for i in range(len(nodes)): dic[nodes[i]] = i edges = [[] for i in range(len(nodes))] for i in range(M): a, b, c = edge_list[i] edges[dic[a]].append((dic[b], c)) for i in range(1, len(nodes)): edges[dic[nodes[i-1]]].append((dic[nodes[i]], 2*(nodes[i]-nodes[i-1]))) from heapq import heappush, heappop INF = 10 ** 18 def dijkstra(s, n, connect): #(始点, ノード数) distance = [INF] * n que = [(0, s)] #(distance, node) distance[s] = 0 confirmed = [False] * n # ノードが確定済みかどうか while que: w,v = heappop(que) if distance[v]<w: continue confirmed[v] = True for to, cost in connect[v]: # ノード v に隣接しているノードに対して if confirmed[to] == False and distance[v] + cost < distance[to]: distance[to] = distance[v] + cost heappush(que, (distance[to], to)) return distance distance = dijkstra(0, len(nodes), edges) ans = distance[-1] print(ans)