結果

問題 No.20 砂漠のオアシス
ユーザー tnodinotnodino
提出日時 2022-02-12 15:32:36
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 489 ms / 5,000 ms
コード長 931 bytes
コンパイル時間 183 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 13,568 KB
最終ジャッジ日時 2024-06-28 17:40:22
合計ジャッジ時間 3,135 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 32 ms
10,752 KB
testcase_01 AC 33 ms
10,752 KB
testcase_02 AC 31 ms
10,880 KB
testcase_03 AC 42 ms
10,752 KB
testcase_04 AC 53 ms
11,008 KB
testcase_05 AC 489 ms
13,568 KB
testcase_06 AC 65 ms
11,520 KB
testcase_07 AC 392 ms
12,672 KB
testcase_08 AC 124 ms
11,520 KB
testcase_09 AC 327 ms
12,160 KB
testcase_10 AC 29 ms
10,752 KB
testcase_11 AC 30 ms
10,752 KB
testcase_12 AC 40 ms
10,752 KB
testcase_13 AC 39 ms
10,752 KB
testcase_14 AC 58 ms
10,880 KB
testcase_15 AC 52 ms
10,752 KB
testcase_16 AC 84 ms
10,880 KB
testcase_17 AC 72 ms
10,880 KB
testcase_18 AC 73 ms
11,136 KB
testcase_19 AC 83 ms
10,880 KB
testcase_20 AC 36 ms
10,752 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

from heapq import heappop, heappush
def Dijkstra(H, W, L, sx, sy, gx, gy, V, Ox, Oy):
    def _Dijkstra():
        while heap:
            C,x,y = heappop(heap)
            C *= -1
            if Cost[x][y] >= C:
                continue
            Cost[x][y] = C
            for i in range(len(dx)):
                nx,my = x + dx[i], y + dy[i]
                if nx < 0 or H <= nx or my < 0 or W <= my:
                    continue
                heappush(heap, (-(C-L[nx][my]), nx, my))
    INF = -1
    dx = [0, 0, -1, 1]
    dy = [-1, 1, 0, 0]
    Cost = [[INF] * W for _ in range(H)]
    heap = []
    heappush(heap, (-V, sx, sy))
    _Dijkstra()
    heappush(heap, (-(Cost[Ox][Oy]*2), Ox, Oy))
    _Dijkstra()
    return Cost[gx][gy]

N,V,Oy,Ox = map(int,input().split())
L = [list(map(int,input().split())) for _ in range(N)]
if Dijkstra(N, N, L, 0, 0, N-1, N-1, V, Ox-1, Oy-1) > 0:
    print('YES')
else:
    print('NO')
0