結果

問題 No.1638 Robot Maze
コンテスト
ユーザー qib
提出日時 2022-12-07 19:43:10
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 971 bytes
コンパイル時間 243 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 78,632 KB
最終ジャッジ日時 2024-10-14 00:19:23
合計ジャッジ時間 6,672 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 47 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

import heapq

INF = 1 << 60

h, w = map(int, input().split())
u, d, r, l, k, p = map(int, input().split())
xs, ys, xt, yt = map(int, input().split())
xs -= 1
ys -= 1
xt -= 1
yt -= 1

c = [input() for _ in range(h)]
dist = [[[INF for _ in range(2)] for _ in range(w)] for _ in range(h)]
dist[xs][ys][0] = 0

dx = [1, 0, -1, 0]
dy = [0, 1, 0, -1]
cost = [[r, r + p], [u, u + p], [l, l + p], [r, r + p]]

hp = [(0, xs, ys, 0)]
while len(hp) > 0:
  cd, cx, cy, ci = heapq.heappop(hp)
  if dist[cx][cy][ci] < cd:
    continue
  for d in range(4):
    nx = cx + dx[d]
    if nx < 0 or nx >= h:
      continue
    ny = cy + dy[d]
    if ny < 0 or ny >= w:
      continue
    if c[nx][ny] == '#':
      continue
    ni = 1 if c[nx][ny] == '@' else 0

    if dist[nx][ny][ni] > dist[cx][cy][ci] + cost[d][ni]:
      dist[nx][ny][ni] = dist[cx][cy][ci] + cost[d][ni]
      heapq.heappush(hp, (dist[nx][ny][ni], nx, ny, ni))

ans = "Yes" if dist[xt][yt][0] <= k else "No"
print(ans)
0