結果

問題 No.424 立体迷路
コンテスト
ユーザー takakin
提出日時 2020-05-07 17:29:11
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 58 ms / 2,000 ms
コード長 967 bytes
コンパイル時間 75 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 11,008 KB
最終ジャッジ日時 2024-07-03 09:31:52
合計ジャッジ時間 1,826 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input=lambda: sys.stdin.readline().rstrip()
h,w=map(int,input().split())
sx,sy,gx,gy=map(int,input().split())
B=[input() for _ in range(h)]
n=h*w
d=((1,0),(0,1),(-1,0),(0,-1))

par=[-1]*n

def find(x):
  if par[x]<0:
    return x
  else:
    par[x]=find(par[x])
    return par[x]

def unite(x,y):
  x=find(x)
  y=find(y)
  if x==y:
    return False
  else:
    if par[x]>par[y]:
      x,y=y,x
    par[x]+=par[y]
    par[y]=x
    return True

def same(x,y):
  return find(x)==find(y)

def size(x):
  return -par[find(x)]

for i in range(h):
  for j in range(w):
    cur=int(B[i][j])
    for dx,dy in d:
      if 0<=i+dx<h and 0<=j+dy<w:
        if abs(cur-int(B[i+dx][j+dy]))<=1:
          unite(i*w+j,(i+dx)*w+j+dy)
    for dx,dy in d:
      if 0<=i+2*dx<h and 0<=j+2*dy<w:
        if cur==int(B[i+2*dx][j+2*dy]) and int(B[i+dx][j+dy])<cur:
          unite(i*w+j,(i+2*dx)*w+j+2*dy)


if same((sx-1)*w+sy-1,(gx-1)*w+gy-1):
  print("YES")
else:
  print("NO")
0