結果

問題 No.20 砂漠のオアシス
コンテスト
ユーザー ckawatak
提出日時 2018-12-17 23:03:47
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,034 bytes
コンパイル時間 149 ms
コンパイル使用メモリ 82,108 KB
実行使用メモリ 85,552 KB
最終ジャッジ日時 2024-09-25 07:38:46
合計ジャッジ時間 14,227 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 20 WA * 1
権限があれば一括ダウンロードができます

ソースコード

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([float('inf') for _ in range(N)])

def dijkstra(sx,sy):
    dx = [1, 0, -1, 0]
    dy = [0, 1, 0, -1]
    
    distance[sy][sx] = 0    
    que = PriorityQueue()
    que.put((sx,sy,0))
    
    while not que.empty():
        x,y,v = 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((nx,ny,nv))

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