結果
問題 | No.1638 Robot Maze |
ユーザー |
|
提出日時 | 2022-12-07 19:45:09 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 181 ms / 2,000 ms |
コード長 | 971 bytes |
コンパイル時間 | 195 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 79,220 KB |
最終ジャッジ日時 | 2024-10-14 00:20:27 |
合計ジャッジ時間 | 7,222 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 49 |
ソースコード
import heapqINF = 1 << 60h, w = map(int, input().split())u, d, r, l, k, p = map(int, input().split())xs, ys, xt, yt = map(int, input().split())xs -= 1ys -= 1xt -= 1yt -= 1c = [input() for _ in range(h)]dist = [[[INF for _ in range(2)] for _ in range(w)] for _ in range(h)]dist[xs][ys][0] = 0dx = [1, 0, -1, 0]dy = [0, 1, 0, -1]cost = [[d, d + p], [r, r + p], [u, u + p], [l, l + p]]hp = [(0, xs, ys, 0)]while len(hp) > 0:cd, cx, cy, ci = heapq.heappop(hp)if dist[cx][cy][ci] < cd:continuefor d in range(4):nx = cx + dx[d]if nx < 0 or nx >= h:continueny = cy + dy[d]if ny < 0 or ny >= w:continueif c[nx][ny] == '#':continueni = 1 if c[nx][ny] == '@' else 0if 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)