結果

問題 No.34 砂漠の行商人
コンテスト
ユーザー rlangevin
提出日時 2023-09-19 12:18:08
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 1,611 bytes
コンパイル時間 540 ms
コンパイル使用メモリ 81,904 KB
実行使用メモリ 666,396 KB
最終ジャッジ日時 2024-07-05 21:41:04
合計ジャッジ時間 10,477 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 19 MLE * 1 -- * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import deque
inf = 10 ** 18
def bfs(G, s, N, N2):
    Q = deque([])
    dist = [inf] * N
    par = [-1] * N
    sz = [1] * N
    val = [[-1] * N2 for i in range(N2)] 
    dist[s] = 0
    for u in G[s]:
        par[u] = s
        dist[u] = 1
        sz[u] += sz[s]
        Q.append(u)
 
    while Q:
        u = Q.popleft()
        for v in G[u]:
            if dist[v] != inf:
                continue
            x, y, vv = g(v)
            if val[x][y] > vv:
                continue
            dist[v] = dist[u] + 1
            par[v] = u
            sz[v] += sz[u]
            Q.append(v)
            
    return dist


N, V, sx, sy, gx, gy = map(int, input().split())
sx, sy, gx, gy = sx-1, sy-1, gx-1, gy-1
d = abs(sx - gx) + abs(sy - gy)
L = []
for i in range(N):
    L.append(list(map(int, input().split())))

M = N * N * (V + 1)
if V > d * 9:
    print(d)
    exit()

def f(x, y, v):
    return x + y * N + v * N * N

def g(n):
    v = n // (N * N)
    n %= N * N
    y, x = divmod(n, N)
    return x, y, v 

dx = [1, 0, -1, 0]
dy = [0, 1, 0, -1]
G = [[] for i in range(M)]
for i in range(N):
    for j in range(N):
        for k in range(4):
            x = i + dx[k]
            y = j + dy[k]
            if x < 0 or x > N - 1 or y < 0 or y > N - 1:
                continue
            for v in range(1, V + 1):
                if v - L[x][y] <= 0:
                    continue
                G[f(i, j, v)].append(f(x, y, v - L[x][y]))

D = bfs(G, f(sy, sx, V), M, N)
ans = inf
for v in range(1, V + 1):
    ans = min(ans, D[f(gy, gx, v)])

print(ans) if ans != inf else print(-1)
0