結果

問題 No.20 砂漠のオアシス
ユーザー Navier_BoltzmannNavier_Boltzmann
提出日時 2022-10-11 04:18:36
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 268 ms / 5,000 ms
コード長 1,427 bytes
コンパイル時間 175 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 87,368 KB
最終ジャッジ日時 2024-06-25 01:01:56
合計ジャッジ時間 4,264 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 52 ms
55,680 KB
testcase_01 AC 55 ms
56,448 KB
testcase_02 AC 52 ms
56,064 KB
testcase_03 AC 125 ms
77,440 KB
testcase_04 AC 123 ms
77,568 KB
testcase_05 AC 258 ms
86,780 KB
testcase_06 AC 264 ms
87,252 KB
testcase_07 AC 268 ms
87,368 KB
testcase_08 AC 188 ms
86,400 KB
testcase_09 AC 261 ms
86,912 KB
testcase_10 AC 52 ms
55,808 KB
testcase_11 AC 54 ms
55,808 KB
testcase_12 AC 140 ms
78,208 KB
testcase_13 AC 142 ms
78,012 KB
testcase_14 AC 160 ms
78,720 KB
testcase_15 AC 149 ms
78,464 KB
testcase_16 AC 188 ms
79,680 KB
testcase_17 AC 176 ms
79,360 KB
testcase_18 AC 177 ms
79,736 KB
testcase_19 AC 190 ms
80,200 KB
testcase_20 AC 120 ms
77,568 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import *
from itertools import *
from functools import *
from heapq import *
import sys,math
input = sys.stdin.readline

N,V,X,Y = map(int,input().split())
L = [list(map(int,input().split())) for _ in range(N)]

M = N**2
e = [[] for _ in range(M)]

def f(i,j):
    
    return N*i+j

H = N
W = N
def nb(x,y):
    tmp = []
    if x+1<H:
        tmp.append((x+1,y))
    if x-1>=0:
        tmp.append((x-1,y))
    if y+1<W:
        tmp.append((x,y+1))
    if y-1>=0:
        tmp.append((x,y-1))
    return tmp
    
X -= 1
Y -= 1
X,Y = Y,X
for i in range(N):
    for j in range(N):
        
        x = f(i,j)
        
        for ix,iy in nb(i,j):
            
            e[x].append((f(ix,iy),L[ix][iy]))
            
INF = (1<<30)
def dijkstra(s,e):
    
    N = len(e)
    dist = [INF]*N
    dist[s]=0
    h = []
    heappush(h,(0,s))
    while h:
        nw,v = heappop(h)
        if dist[v]!=nw:
            continue
        for iv,ic in e[v]:
            nc = ic + nw
            if nc < dist[iv]:
                dist[iv] = nc
                heappush(h,(nc,iv))
    return dist



D = dijkstra(0,e)
if (X,Y)==(-1,-1):
    if D[-1]<V:
        print('YES')
    else:
        print('NO')
    exit()
           
if D[-1]<V:
    print('YES')
    exit()   
oasis = f(X,Y)
if D[oasis]>=V:
    print('NO')
    exit()
H = V - D[oasis]
H *= 2
D =dijkstra(oasis,e)
if D[-1]<H:
    print('YES')
else:
    print('NO')
0