結果
問題 | No.2695 Warp Zone |
ユーザー | miya145592 |
提出日時 | 2024-03-22 22:56:33 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,552 ms / 2,000 ms |
コード長 | 1,795 bytes |
コンパイル時間 | 289 ms |
コンパイル使用メモリ | 81,788 KB |
実行使用メモリ | 319,512 KB |
最終ジャッジ日時 | 2024-09-30 12:19:03 |
合計ジャッジ時間 | 19,647 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 |
ソースコード
from heapq import heappush, heappop def dijkstra(s, N): # (始点, ノード数) dist = [INF for _ in range(N)] hq = [(0, s)] dist[s] = 0 seen = [False] * N # ノードが確定済みかどうか while hq: d, v = heappop(hq) # ノードを pop する if seen[v]: continue seen[v] = True for to, cost in G[v]: # ノード v に隣接しているノードに対して if dist[v] + cost < dist[to]: dist[to] = dist[v] + cost heappush(hq, (dist[to], to)) return dist import sys input = sys.stdin.readline H, W, N = map(int, input().split()) ABCD = [list(map(int, input().split())) for _ in range(N)] G = [[] for _ in range(2*N+2)] INF = H+W st = 2*N ed = 2*N+1 G[st].append((ed, H-1+W-1)) G[ed].append((st, H-1+W-1)) for i in range(N): a, b, c, d = ABCD[i] u = i*2 v = i*2+1 G[u].append((v, 1)) G[v].append((u, abs(c-a)+abs(d-b))) G[st].append((u, a-1+b-1)) G[u].append((st, a-1+b-1)) G[ed].append((u, H-a+W-b)) G[u].append((ed, H-a+W-b)) G[st].append((v, c-1+d-1)) G[v].append((st, c-1+d-1)) G[ed].append((v, H-c+W-d)) G[v].append((ed, H-c+W-d)) for i in range(N): a, b, c, d = ABCD[i] u = i*2 v = i*2+1 for j in range(i+1, N): a2, b2, c2, d2 = ABCD[j] u2 = j*2 v2 = j*2+1 G[u].append((u2, abs(a2-a)+abs(b2-b))) G[u2].append((u, abs(a2-a)+abs(b2-b))) G[u].append((v2, abs(c2-a)+abs(d2-b))) G[v2].append((u, abs(c2-a)+abs(d2-b))) G[v].append((u2, abs(a2-c)+abs(b2-d))) G[u2].append((v, abs(a2-c)+abs(b2-d))) G[v].append((v2, abs(c2-c)+abs(d2-d))) G[v2].append((v, abs(c2-c)+abs(d2-d))) dist = dijkstra(st, 2*N+2) print(dist[ed])