結果

問題 No.1949 足し算するだけのパズルゲーム(2)
ユーザー Akijin_007
提出日時 2022-05-20 23:48:33
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 981 ms / 3,000 ms
コード長 1,152 bytes
コンパイル時間 176 ms
コンパイル使用メモリ 82,556 KB
実行使用メモリ 99,776 KB
最終ジャッジ日時 2024-09-20 10:14:21
合計ジャッジ時間 8,942 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #

#int(input())
#map(int, input().split())
#list(map(int, input().split()))

H, W, Y, X = map(int, input().split())
X -= 1
Y -= 1

A = [0] * H
for i in range(H):
    A[i] = list(map(int, input().split()))

import heapq

q = []
# print(A[Y])
s = A[Y][X]

v = [[0] * W for i in range(H)]
v[Y][X] = 1

if X + 1 < W:
    q.append([A[Y][X+1], Y, X+1])
    v[Y][X+1] = 1
if X - 1 >= 0:
    q.append([A[Y][X-1], Y, X-1])
    v[Y][X-1] = 1
if Y + 1 < H:
    q.append([A[Y+1][X], Y+1, X])
    v[Y+1][X] = 1
if Y - 1 >= 0:
    q.append([A[Y-1][X], Y-1, X])
    v[Y-1][X] = 1

heapq.heapify(q)


dx = [1, -1, 0, 0]
dy = [0, 0, 1, -1]

f = 1

while len(q) > 0:
    t = heapq.heappop(q)
    a = t[0]
    x = t[1]
    y = t[2]
    if s <= a:
        f = 0
        break
    s += a
    for i in range(4):
        nx = x + dx[i]
        ny = y + dy[i]
        # print(nx, ny)
        if nx < 0 or nx >= H or ny < 0 or ny >= W:
            continue
        if v[nx][ny] == 1:
            continue
        heapq.heappush(q, [A[nx][ny], nx, ny])
        v[nx][ny] = 1

# for i in range(H):
#     print(v[i])
# print(q)


if f == 0:
    print("No")
else:
    print("Yes")


0