結果

問題 No.20 砂漠のオアシス
コンテスト
ユーザー nagitaosu
提出日時 2020-03-13 14:13:07
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,706 bytes
コンパイル時間 105 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 32,896 KB
最終ジャッジ日時 2024-11-22 01:48:22
合計ジャッジ時間 3,843 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 16 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env python3
import sys
input = sys.stdin.readline
import heapq
INF = 10**9

class Dijkstra:
    def __init__(self, adj):
        self.adj = adj
        self.dist = [INF] * len(adj)
        self.q = []

    def reset(self):
        self.dist = [INF] * len(self.adj)
        self.q = []

    def calc(self, start):
        self.dist[start] = 0
        heapq.heappush(self.q, (0, start))
        while len(self.q) != 0:
            prov_cost, src = heapq.heappop(self.q)
            if self.dist[src] < prov_cost:
                continue
            for dest, cost in self.adj[src]:
                if self.dist[dest] > self.dist[src] + cost:
                    self.dist[dest] = self.dist[src] + cost
                    heapq.heappush(self.q, (self.dist[dest], dest))
        return self.dist

n, v, ox, oy = map(int, input().split())
ox -= 1; oy -= 1
edge = [[] for _ in range(n * n)]

field = []
for _ in range(n):
    field.append([int(item) for item in input().split()])

delta = [(0, 1), (1, 0), (0, -1), (-1, 0)]
for i in range(n):
    for j in range(n):
        for dx, dy in delta:
            if i + dx < 0 or i + dx >= n:
                continue
            if j + dy < 0 or j + dy >= n:
                continue
            frm = i * n + j 
            too = (i + dx) * n + (j + dy)
            edge[frm].append((too, field[i + dx][j + dy]))

DIJK = Dijkstra(edge)
dist_from_start = DIJK.calc(0)
ans = v - dist_from_start[n*n-1]
if ox != -1 and oy != -1:
    DIJK.reset()
    dist_from_oasis = DIJK.calc(ox*n + oy)
    ret = v - dist_from_start[ox * n + oy]
    ret *= 2
    ret -= dist_from_oasis[n*n - 1]
    ans = max(ans, ret)

if ans > 0:
    print("YES")
else:
    print("NO")
0