結果

問題 No.34 砂漠の行商人
ユーザー roaris
提出日時 2020-07-25 22:59:47
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
MLE  
実行時間 -
コード長 1,234 bytes
コンパイル時間 349 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 694,876 KB
最終ジャッジ日時 2024-06-27 18:13:24
合計ジャッジ時間 20,971 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 17 MLE * 1 -- * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
from collections import deque

def bfs():
    dist = [-1]*c
    dist[serial[Sx][Sy][V]] = 0
    q = deque([serial[Sx][Sy][V]])
    
    while q:
        v = q.popleft()
        
        for nv in G[v]:
            if dist[nv]==-1:
                dist[nv] = dist[v]+1
                q.append(nv)
    
    return dist

N, V, Sy, Sx, Gy, Gx = map(int, input().split())
Sy -= 1
Sx -= 1
Gy -= 1
Gx -= 1
L = [list(map(int, input().split())) for _ in range(N)]

if V>=1800:
    print(abs(Sx-Gx)+abs(Sy-Gy))
    exit()

serial = [[[-1]*(V+1) for _ in range(N)] for _ in range(N)]
c = 0

for i in range(N):
    for j in range(N):
        for k in range(V+1):
            serial[i][j][k] = c
            c += 1

G = [[] for _ in range(c)]

for i in range(N):
    for j in range(N):
        for k in range(V+1):
            for ni, nj in [(i-1, j), (i+1, j), (i, j-1), (i, j+1)]:
                if 0<=ni<N and 0<=nj<N and k-L[ni][nj]>0:
                    G[serial[i][j][k]].append(serial[ni][nj][k-L[ni][nj]])

dist = bfs()
ans = 10**18

for i in range(V+1):
    if dist[serial[Gx][Gy][i]]!=-1:
        ans = min(ans, dist[serial[Gx][Gy][i]])
        
if ans==10**18:
    print(-1)
else:
    print(ans)
0