結果

問題 No.34 砂漠の行商人
ユーザー tktk_snsn
提出日時 2021-05-17 19:56:38
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 962 bytes
コンパイル時間 210 ms
コンパイル使用メモリ 82,412 KB
実行使用メモリ 848,196 KB
最終ジャッジ日時 2024-10-06 23:15:41
合計ジャッジ時間 2,923 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 9 MLE * 1 -- * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import deque
import sys
input = sys.stdin.buffer.readline
sys.setrecursionlimit(10 ** 7)
DX = (-1, 0, 1, 0, -1, -1, 1, 1)
DY = (0, 1, 0, -1, -1, 1, -1, 1)
DX = DX[:4]
DY = DY[:4]


N, V, sy, sx, gy, gx = map(int, input().split())
sx -= 1
sy -= 1
gx -= 1
gy -= 1
L = tuple(tuple(map(int, input().split())) for _ in range(N))
Nsq = N*N

dist = [[-1]*(V+1) for _ in range(N*N)]
dist = [-1] * ((V+1)*Nsq)
dist[V*Nsq+sx*N+sy] = 0
que = deque([V*Nsq+sx*N+sy])
ans = -1
while que:
    vxy = que.popleft()
    v, xy = divmod(vxy, Nsq)
    x, y = divmod(xy, N)
    d = dist[vxy]
    if (x, y) == (gx, gy):
        ans = d
        break
    for dx, dy in zip(DX, DY):
        nx = x + dx
        ny = y + dy
        if 0 <= nx < N and 0 <= ny < N and v - L[nx][ny] > 0:
            nv = v - L[nx][ny]
            nvxy = nv * Nsq + nx * N + ny
            if dist[nvxy] == -1:
                dist[nvxy] = d + 1
                que.append(nvxy)

print(ans)
0