結果

問題 No.424 立体迷路
コンテスト
ユーザー nanae
提出日時 2017-01-19 09:00:10
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 66 ms / 2,000 ms
コード長 1,444 bytes
コンパイル時間 153 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 11,392 KB
最終ジャッジ日時 2024-07-05 07:13:18
合計ジャッジ時間 1,514 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

# 入力 
h, w = map(int, input().split())
sx, sy, gx, gy = map(lambda x: x - 1, map(int, input().split()))
# sx, sy, gx, gy = sx - 1, sy - 1, gx - 1, gy - 1

b = [[] for i in range(h)]

for i in range(h):
    line = input()

    for takasa in line:
        b[i].append(int(takasa))

# print(b)

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

ikeru = [[None] * w for i in range(h)]

# print(ikeru)

ikeru[sx][sy] = True
tugis = [(sx, sy)]
tansaku_zumi = set()

# print(tugis[0][0])

while tugis:
    cur = tugis.pop()

    for i in range(len(dx)):
        if 0 <= cur[0] + dx[i] < h and 0 <= cur[1] + dy[i] < w:
            if i > 3 and b[cur[0]][cur[1]] == b[cur[0] + dx[i]][cur[1] + dy[i]]:
                if b[cur[0]][cur[1]] > b[cur[0] + dx[i - 4]][cur[1] + dy[i - 4]]:
                    if (cur[0] + dx[i], cur[1] + dy[i]) not in tansaku_zumi:
                        ikeru[cur[0] + dx[i]][cur[1] + dy[i]] = True
                        tugis.append((cur[0] + dx[i], cur[1] + dy[i]))
            elif i <= 3 and abs(b[cur[0] + dx[i]][cur[1] + dy[i]] - b[cur[0]][cur[1]]) <= 1:
                if (cur[0] + dx[i], cur[1] + dy[i]) not in tansaku_zumi:
                    ikeru[cur[0] + dx[i]][cur[1] + dy[i]] = True
                    tugis.append((cur[0] + dx[i], cur[1] + dy[i]))

    tansaku_zumi.add(cur)

# print(sorted(list(tansaku_zumi)))

if (gx, gy) in tansaku_zumi:
    print('YES')
else:
    print('NO')
0