結果

問題 No.424 立体迷路
ユーザー tokizo
提出日時 2019-02-01 19:19:29
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,584 bytes
コンパイル時間 83 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 145,792 KB
最終ジャッジ日時 2024-11-23 22:53:33
合計ジャッジ時間 8,670 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 17 WA * 2 TLE * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

import pprint
from collections import deque

def main():
    h, w = map(int, input().split())
    sx, sy, gx, gy = map(int, input().split())
    sx -= 1
    sy -= 1
    gx -= 1
    gy -= 1

    B = []
    for i in range(h):
        B.append(list(map(int, list(input()))))

    if(sx == gx and sy == gy):
        print("YES")
        exit()

    seen1 = [[False for _ in range(w)] for _ in range(h)]
    seen2 = [[False for _ in range(w)] for _ in range(h)]

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

    q = deque()
    q.append((sx, sy, 0))
    while q:
        x, y, z = q.popleft()
        if z == 0:
            seen1[x][y] = True
            seen2[x][y] = True
        elif z == 1:
            seen1[x][y] = True
        elif z == 2:
            seen2[x][y] = True
        
        for i in range(8):
            nx = x + dx[i]
            ny = y + dy[i]
            if nx < 0 or nx >= h or ny < 0 or ny >= w:
                continue

            if i < 4:
                if abs(B[x][y] - B[nx][ny]) <= 1 and seen1[nx][ny] == False:
                    if nx == gx and ny == gy:
                        print("YES")
                        exit()
                    else:
                        q.append((nx, ny, 1))
            else:
                if B[x][y] == B[nx][ny] and seen2[nx][ny] == False:
                    if nx == gx and ny == gy:
                        print("YES")
                        exit()
                    else:
                        q.append((nx, ny, 2))
    print("NO")


if __name__ == "__main__":
    main()
0