結果

問題 No.424 立体迷路
ユーザー Coki628Coki628
提出日時 2020-09-23 12:53:56
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 65 ms / 2,000 ms
コード長 2,426 bytes
コンパイル時間 164 ms
コンパイル使用メモリ 82,800 KB
実行使用メモリ 68,864 KB
最終ジャッジ日時 2024-06-27 21:44:03
合計ジャッジ時間 2,428 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 43 ms
54,784 KB
testcase_01 AC 43 ms
54,272 KB
testcase_02 AC 43 ms
54,784 KB
testcase_03 AC 45 ms
54,400 KB
testcase_04 AC 45 ms
54,400 KB
testcase_05 AC 44 ms
54,784 KB
testcase_06 AC 43 ms
54,400 KB
testcase_07 AC 44 ms
54,400 KB
testcase_08 AC 43 ms
54,528 KB
testcase_09 AC 43 ms
54,528 KB
testcase_10 AC 42 ms
54,272 KB
testcase_11 AC 44 ms
54,272 KB
testcase_12 AC 43 ms
54,656 KB
testcase_13 AC 44 ms
54,656 KB
testcase_14 AC 44 ms
54,400 KB
testcase_15 AC 44 ms
54,272 KB
testcase_16 AC 44 ms
54,400 KB
testcase_17 AC 49 ms
61,952 KB
testcase_18 AC 49 ms
61,696 KB
testcase_19 AC 51 ms
62,336 KB
testcase_20 AC 51 ms
61,568 KB
testcase_21 AC 43 ms
54,400 KB
testcase_22 AC 45 ms
54,528 KB
testcase_23 AC 43 ms
54,656 KB
testcase_24 AC 64 ms
68,864 KB
testcase_25 AC 65 ms
68,352 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

def input(): return sys.stdin.readline().strip()
def list2d(a, b, c): return [[c for j in range(b)] for i in range(a)]
def list3d(a, b, c, d): return [[[d for k in range(c)] for j in range(b)] for i in range(a)]
def list4d(a, b, c, d, e): return [[[[e for l in range(d)] for k in range(c)] for j in range(b)] for i in range(a)]
def ceil(x, y=1): return int(-(-x // y))
def INT(): return int(input())
def MAP(): return map(int, input().split())
def LIST(N=None): return list(MAP()) if N is None else [INT() for i in range(N)]
def Yes(): print('Yes')
def No(): print('No')
def YES(): print('YES')
def NO(): print('NO')
sys.setrecursionlimit(10**9)
INF = 10**18
MOD = 10**9 + 7
EPS = 10**-10

def build_grid(H, W, intv, _type, space=False, padding=True):
    # 入力がスペース区切りかどうか
    if space:
        _input = lambda: input().split()
    else:
        _input = lambda: input()
    _list = lambda: list(map(_type, _input()))
    # 余白の有無
    if padding:
        offset = 1
    else:
        offset = 0
    grid = list2d(H+offset*2, W+offset*2, intv)
    for i in range(offset, H+offset):
        row = _list()
        for j in range(offset, W+offset):
            grid[i][j] = row[j-offset]
    return grid

def bfs(grid, src):
    """ BFS(グリッド、重みなし) """
    from collections import deque

    H, W = len(grid), len(grid[0])
    directions = ((1, 0), (-1, 0), (0, 1), (0, -1))
    dist = list2d(H, W, 0)
    que = deque()
    for h, w in src:
        que.append((h, w))
        dist[h][w] = 0
    while que:
        h, w = que.popleft()
        for dh, dw in directions:
            nh = h + dh
            nw = w + dw
            if grid[nh][nw] > grid[h][w]+1 or grid[nh][nw] < grid[h][w]-1:
                continue
            if dist[nh][nw]:
                continue
            dist[nh][nw] = 1
            que.append((nh, nw))
        for dh, dw in directions:
            nh = h + dh
            nw = w + dw
            if grid[nh][nw] > grid[h][w]:
                continue
            nh += dh
            nw += dw
            if grid[nh][nw] != grid[h][w]:
                continue
            if dist[nh][nw]:
                continue
            dist[nh][nw] = 1
            que.append((nh, nw))
    return dist

H, W = MAP()
sh, sw, gh, gw = MAP()
grid = build_grid(H, W, 20, int)

res = bfs(grid, [(sh, sw)])
if res[gh][gw]:
    YES()
else:
    NO()
0