結果

問題 No.20 砂漠のオアシス
ユーザー ckawatak
提出日時 2018-12-18 22:52:57
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

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