結果

問題 No.34 砂漠の行商人
コンテスト
ユーザー roaris
提出日時 2020-07-25 22:56:38
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,234 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 580 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 852,480 KB
最終ジャッジ日時 2024-06-27 18:11:49
合計ジャッジ時間 9,468 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 14 WA * 3 MLE * 1 -- * 8
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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-Sy)+abs(Gx-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