結果

問題 No.20 砂漠のオアシス
ユーザー nagitaosunagitaosu
提出日時 2020-03-13 14:43:24
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 407 ms / 5,000 ms
コード長 1,706 bytes
コンパイル時間 320 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 32,896 KB
最終ジャッジ日時 2024-11-22 02:54:02
合計ジャッジ時間 4,026 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

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, oy, ox = 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")
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0