結果

問題 No.20 砂漠のオアシス
ユーザー ckawatakckawatak
提出日時 2018-12-18 22:52:57
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 442 ms / 5,000 ms
コード長 1,120 bytes
コンパイル時間 259 ms
コンパイル使用メモリ 11,964 KB
実行使用メモリ 13,368 KB
最終ジャッジ日時 2023-10-25 23:45:24
合計ジャッジ時間 4,196 ms
ジャッジサーバーID
(参考情報)
judge12 / judge13
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 33 ms
10,496 KB
testcase_01 AC 33 ms
10,500 KB
testcase_02 AC 33 ms
10,496 KB
testcase_03 AC 44 ms
10,624 KB
testcase_04 AC 45 ms
10,628 KB
testcase_05 AC 390 ms
13,368 KB
testcase_06 AC 438 ms
12,276 KB
testcase_07 AC 440 ms
12,276 KB
testcase_08 AC 442 ms
12,276 KB
testcase_09 AC 439 ms
12,276 KB
testcase_10 AC 32 ms
10,496 KB
testcase_11 AC 32 ms
10,496 KB
testcase_12 AC 48 ms
10,592 KB
testcase_13 AC 52 ms
10,608 KB
testcase_14 AC 69 ms
10,696 KB
testcase_15 AC 58 ms
10,656 KB
testcase_16 AC 113 ms
10,896 KB
testcase_17 AC 89 ms
10,772 KB
testcase_18 AC 101 ms
10,852 KB
testcase_19 AC 114 ms
10,904 KB
testcase_20 AC 42 ms
10,560 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

from queue import PriorityQueue

N,V,Ox,Oy = list(map(int, input().split(' ')))

desert = []
for _ in range(N):
    desert.append(list(map(int, input().split(' '))))

distance = []
for _ in range(N):
    distance.append([0 for _ in range(N)])
    
def dijkstra(sx,sy):
    dx = [1, 0, -1, 0]
    dy = [0, 1, 0, -1]

    for i in range(N):
        for j in range(N):
            distance[i][j] = float('inf')
    
    distance[sy][sx] = 0    
    que = PriorityQueue()
    que.put((0,sx,sy))
    
    while not que.empty():
        v,x,y = que.get()

        for i in range(len(dx)):
            nx = x + dx[i]
            ny = y + dy[i]

            if 0 <= nx and nx < N and 0 <= ny and ny < N:
                nv = v + desert[ny][nx]
                if nv < distance[ny][nx]:
                    distance[ny][nx] = nv
                    que.put((nv,nx,ny))

dijkstra(0,0)
if distance[N-1][N-1] < V:
    print('YES')
    exit(0)
    
if Ox == 0 and Oy == 0:
    print('NO')
    exit(0)
    
ov = (V - distance[Oy-1][Ox-1]) * 2
dijkstra(Ox-1,Oy-1)
if 0 < ov - distance[N-1][N-1]:
    print('YES')
else:
    print('NO')
0