結果

問題 No.424 立体迷路
ユーザー Coki628Coki628
提出日時 2020-09-23 12:53:56
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 118 ms / 2,000 ms
コード長 2,426 bytes
コンパイル時間 308 ms
コンパイル使用メモリ 87,804 KB
実行使用メモリ 78,688 KB
最終ジャッジ日時 2023-09-10 06:14:26
合計ジャッジ時間 4,629 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 95 ms
72,244 KB
testcase_01 AC 95 ms
71,964 KB
testcase_02 AC 95 ms
72,144 KB
testcase_03 AC 95 ms
72,212 KB
testcase_04 AC 95 ms
72,176 KB
testcase_05 AC 96 ms
71,988 KB
testcase_06 AC 94 ms
72,384 KB
testcase_07 AC 97 ms
72,136 KB
testcase_08 AC 95 ms
72,256 KB
testcase_09 AC 96 ms
72,104 KB
testcase_10 AC 98 ms
72,216 KB
testcase_11 AC 97 ms
71,980 KB
testcase_12 AC 97 ms
71,840 KB
testcase_13 AC 96 ms
72,008 KB
testcase_14 AC 98 ms
72,100 KB
testcase_15 AC 99 ms
72,360 KB
testcase_16 AC 95 ms
71,988 KB
testcase_17 AC 106 ms
77,948 KB
testcase_18 AC 104 ms
78,072 KB
testcase_19 AC 103 ms
77,916 KB
testcase_20 AC 105 ms
78,000 KB
testcase_21 AC 96 ms
72,188 KB
testcase_22 AC 95 ms
72,012 KB
testcase_23 AC 95 ms
72,176 KB
testcase_24 AC 118 ms
78,688 KB
testcase_25 AC 118 ms
78,444 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