結果

問題 No.1638 Robot Maze
ユーザー AEn
提出日時 2022-06-18 01:08:07
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 165 ms / 2,000 ms
コード長 1,441 bytes
コンパイル時間 393 ms
コンパイル使用メモリ 82,440 KB
実行使用メモリ 79,628 KB
最終ジャッジ日時 2024-10-09 11:41:03
合計ジャッジ時間 6,906 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

from heapq import heappush, heappop
INF = float('inf')
x = [1, 0, 0, -1]
y = [0, 1, -1, 0]
H, W = map(int, input().split())
U, D, R, L, K, P = map(int, input().split())
cost = [D, R, L, U]
sx, sy, gx, gy = map(int, input().split())
sx-=1;sy-=1;gx-=1;gy-=1
C = [list(input()) for _ in range(H)]
adj = [list() for _ in range(H*W)]
for i in range(H):
for j in range(W):
if C[i][j] == '#':
continue
else:
for cc, xx, yy in zip(cost, x, y):
cx, cy = i+xx, j+yy
if 0<=cx<H and 0<=cy<W:
if C[cx][cy] == '#':
continue
elif C[cx][cy] == '.':
adj[i*H+j].append((cx*H+cy, cc))
else:
adj[i*H+j].append((cx*H+cy, P+cc))
def dijkstra(s, n):
dist = [INF] * n
hq = [(0, s)] # (distance, node)
dist[s] = 0
seen = [False] * n #
#
# prev = [-1]*n
while hq:
dis, v = heappop(hq)
if dist[v] < dis:
continue
seen[v] = True
for to, cost in adj[v]:
if seen[to] == False and dist[v] + cost < dist[to]:
dist[to] = dist[v] + cost
# prev[to] = v
heappush(hq, (dist[to], to))
return dist
d = dijkstra(sx*H+sy, H*W)
if d[gx*H+gy]<=K:
print('Yes')
else:
print('No')
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0