結果

問題 No.20 砂漠のオアシス
ユーザー ckawatakckawatak
提出日時 2018-12-18 22:52:57
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 475 ms / 5,000 ms
コード長 1,120 bytes
コンパイル時間 194 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 14,080 KB
最終ジャッジ日時 2024-09-25 07:49:18
合計ジャッジ時間 4,199 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 34 ms
11,136 KB
testcase_01 AC 34 ms
11,264 KB
testcase_02 AC 34 ms
11,264 KB
testcase_03 AC 47 ms
11,520 KB
testcase_04 AC 47 ms
11,264 KB
testcase_05 AC 417 ms
14,080 KB
testcase_06 AC 471 ms
13,056 KB
testcase_07 AC 468 ms
13,056 KB
testcase_08 AC 475 ms
13,184 KB
testcase_09 AC 471 ms
13,056 KB
testcase_10 AC 33 ms
11,264 KB
testcase_11 AC 33 ms
11,264 KB
testcase_12 AC 51 ms
11,392 KB
testcase_13 AC 56 ms
11,264 KB
testcase_14 AC 72 ms
11,392 KB
testcase_15 AC 61 ms
11,264 KB
testcase_16 AC 118 ms
11,648 KB
testcase_17 AC 91 ms
11,392 KB
testcase_18 AC 107 ms
11,520 KB
testcase_19 AC 122 ms
11,648 KB
testcase_20 AC 44 ms
11,264 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