結果

問題 No.20 砂漠のオアシス
ユーザー H20H20
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 45 ms
54,400 KB
testcase_01 AC 52 ms
55,680 KB
testcase_02 AC 48 ms
55,040 KB
testcase_03 AC 154 ms
79,620 KB
testcase_04 AC 214 ms
80,788 KB
testcase_05 AC 532 ms
114,532 KB
testcase_06 AC 577 ms
120,524 KB
testcase_07 AC 602 ms
119,532 KB
testcase_08 AC 611 ms
119,404 KB
testcase_09 AC 567 ms
119,272 KB
testcase_10 AC 46 ms
54,656 KB
testcase_11 AC 46 ms
54,400 KB
testcase_12 AC 185 ms
80,396 KB
testcase_13 AC 174 ms
81,284 KB
testcase_14 AC 226 ms
81,540 KB
testcase_15 AC 207 ms
80,996 KB
testcase_16 AC 308 ms
87,964 KB
testcase_17 AC 254 ms
83,760 KB
testcase_18 AC 268 ms
85,364 KB
testcase_19 AC 284 ms
86,848 KB
testcase_20 AC 136 ms
78,496 KB
権限があれば一括ダウンロードができます

ソースコード

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