結果

問題 No.20 砂漠のオアシス
ユーザー tktk_snsntktk_snsn
提出日時 2020-12-24 04:20:42
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 234 ms / 5,000 ms
コード長 1,060 bytes
コンパイル時間 199 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 79,232 KB
最終ジャッジ日時 2024-09-21 16:47:23
合計ジャッジ時間 3,348 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 41 ms
52,992 KB
testcase_01 AC 40 ms
53,120 KB
testcase_02 AC 41 ms
52,992 KB
testcase_03 AC 89 ms
77,008 KB
testcase_04 AC 95 ms
77,056 KB
testcase_05 AC 234 ms
79,232 KB
testcase_06 AC 101 ms
78,024 KB
testcase_07 AC 164 ms
78,316 KB
testcase_08 AC 123 ms
77,952 KB
testcase_09 AC 156 ms
78,232 KB
testcase_10 AC 40 ms
52,608 KB
testcase_11 AC 39 ms
52,480 KB
testcase_12 AC 81 ms
74,880 KB
testcase_13 AC 88 ms
76,416 KB
testcase_14 AC 102 ms
76,928 KB
testcase_15 AC 95 ms
76,896 KB
testcase_16 AC 107 ms
77,216 KB
testcase_17 AC 108 ms
77,428 KB
testcase_18 AC 105 ms
76,944 KB
testcase_19 AC 107 ms
76,672 KB
testcase_20 AC 75 ms
71,936 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import heapq
import sys
input = sys.stdin.buffer.readline
sys.setrecursionlimit(10 ** 7)
inf = 10000
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, oy, ox = map(int, input().split())
L = tuple(tuple(map(int, input().split())) for _ in range(N))
ox -= 1
oy -= 1

dist = [[inf] * N for _ in range(N)]
dist[0][0] = 0
que = [(0, 0, 0)]
while que:
    ds, x, y = heapq.heappop(que)
    if dist[x][y] < ds:
        continue
    for dx, dy in zip(DX, DY):
        nx = x + dx
        ny = y + dy
        if nx < 0 or N <= nx or ny < 0 or N <= ny:
            continue
        cost = ds + L[nx][ny]
        if cost >= V:
            continue
        if dist[nx][ny] > cost:
            dist[nx][ny] = cost
            if (nx, ny) == (ox, oy):
                dist[nx][ny] -= (V - cost)
                ox = -1
                oy = -1
            heapq.heappush(que, (dist[nx][ny], nx, ny))

# if 1:
#     import numpy as np
#     print(np.array(dist))

if dist[N - 1][N - 1] < V:
    print("YES")
else:
    print("NO")
0