結果

問題 No.20 砂漠のオアシス
ユーザー H20
提出日時 2021-12-21 15:12:37
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 611 ms / 5,000 ms
コード長 1,542 bytes
コンパイル時間 353 ms
コンパイル使用メモリ 81,792 KB
実行使用メモリ 120,524 KB
最終ジャッジ日時 2024-09-15 15:35:25
合計ジャッジ時間 6,657 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge6
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

import collections
import heapq


class Dijkstra:
    def __init__(self):
        self.e = collections.defaultdict(list)

    def add(self, u, v, d):
        self.e[u].append([v, d])

    def delete(self, u, v):
        self.e[u] = [_ for _ in self.e[u] if _[0] != v]
        self.e[v] = [_ for _ in self.e[v] if _[0] != u]

    def search(self, s):
        """
        :param s: 始点
        :return: 始点から各点までの最短経路
        """
        d = collections.defaultdict(lambda: float('inf'))
        d[s] = 0
        q = []
        heapq.heappush(q, (0, s))
        v = collections.defaultdict(bool)
        while len(q):
            k, u = heapq.heappop(q)
            if v[u]:
                continue
            v[u] = True

            for uv, ud in self.e[u]:
                if v[uv]:
                    continue
                vd = k + ud
                if d[uv] > vd:
                    d[uv] = vd
                    heapq.heappush(q, (vd, uv))

        return d

N,V,X,Y = map(int,input().split())
L = []
X,Y=X-1,Y-1
for _ in range(N):
    L.append(list(map(int,input().split())))
graph = Dijkstra()

for i in range(N):
    for j in range(N-1):
        graph.add((i,j),(i,j+1),L[i][j+1])
        graph.add((i,j+1),(i,j),L[i][j])

for i in range(N-1):
    for j in range(N):
        graph.add((i,j),(i+1,j),L[i+1][j])
        graph.add((i+1,j),(i,j),L[i][j])
G = graph.search((0,0))
GO = graph.search((Y,X))
if G[(N-1,N-1)]<V or G[(Y,X)]+GO[(N-1,N-1)]<V+(V-G[(Y,X)]):
    print('YES')
else:
    print('NO')
0