結果
問題 |
No.1638 Robot Maze
|
ユーザー |
|
提出日時 | 2023-08-06 20:58:57 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,040 bytes |
コンパイル時間 | 557 ms |
コンパイル使用メモリ | 82,128 KB |
実行使用メモリ | 78,320 KB |
最終ジャッジ日時 | 2024-11-07 00:06:03 |
合計ジャッジ時間 | 5,089 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 33 WA * 16 |
ソースコード
from heapq import heappop, heappush h, w = map(int, input().split()) u, d, r, l, k, p = map(int, input().split()) si, sj, ti, tj = map(int, input().split()) si -= 1 sj -= 1 ti -= 1 tj -= 1 S = [list(input()) for _ in range(h)] DP = [[[k + 1 for _ in range(2)] for _ in range(w)] for _ in range(h)] Directions = [(-1, 0), (1, 0), (0, 1), (-1, 0)] C = [u, d, r, l] DP[si][sj][0] = 0 H = [(0, si, sj, 0)] while H: cc, ci, cj, ck = heappop(H) if cc > DP[ci][cj][ck]: continue for i in range(4): nc = cc di, dj = Directions[i] ni, nj = ci + di, cj + dj if 0 <= ni < h and 0 <= nj < w: if S[ni][nj] == '#': continue elif S[ni][nj] == '@': nc += p nk = 1 else: nk = 0 nc += C[i] if nc >= DP[ni][nj][nk]: continue DP[ni][nj][nk] = nc heappush(H, (nc, ni, nj, nk)) if DP[ti][tj][0] == k + 1: print("No") else: print("Yes")