結果

問題 No.367 ナイトの転身
コンテスト
ユーザー Navier_Boltzmann
提出日時 2022-09-02 10:37:12
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 428 ms / 2,000 ms
コード長 1,892 bytes
コンパイル時間 213 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 99,884 KB
最終ジャッジ日時 2024-11-15 15:51:08
合計ジャッジ時間 4,350 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

import pypyjit
pypyjit.set_param('max_unroll_recursion=-1')
H,W = map(int,input().split())
S = [list(input()) for _ in range(H)]
for i in range(H):
    for j in range(W):
        if S[i][j] == 'S':
            s = (i,j)
        if S[i][j] == 'G':
            g = (i,j)
            

from collections import deque
v = deque()

dist = [[[-1]*W for _ in range(H)] for j in range(2)]

v.append((0,s[0],s[1]))
dist[0][s[0]][s[1]]=0


def nb1(i,j):
    dx = [1,1,2,2,-1,-1,-2,-2]
    dy = [2,-2,1,-1,2,-2,1,-1]
    tmp = []
    for t in range(8):
        tx = dx[t]+i
        ty = dy[t]+j
        if (0<=tx<H)&(0<=ty<W):
            tmp.append((tx,ty))
    return tmp
 
 
def nb2(i,j):
    dx = [1,1,-1,-1]
    dy = [1,-1,1,-1]
    tmp = []
    for t in range(4):
        tx = dx[t]+i
        ty = dy[t]+j
        if (0<=tx<H)&(0<=ty<W):
            tmp.append((tx,ty))
    return tmp   


while v:
    
    ic,ix,iy = v.popleft()
    td = dist[ic][ix][iy]
    
    if ic==0:
        for jx,jy in nb1(ix,iy):
            
            if S[jx][jy]=='R':
                
                if dist[1][jx][jy]==-1:
                    dist[1][jx][jy] = td+1
                    v.append((1,jx,jy))
            else:
                if dist[0][jx][jy]==-1:
                    dist[0][jx][jy] = td+1
                    v.append((0,jx,jy))        
    else:
        
        for jx,jy in nb2(ix,iy):
            
            if S[jx][jy]=='R':
                
                if dist[0][jx][jy]==-1:
                    dist[0][jx][jy] = td+1
                    v.append((0,jx,jy))
            else:
                if dist[1][jx][jy]==-1:
                    dist[1][jx][jy] = td+1
                    v.append((1,jx,jy))            
                
a = dist[0][g[0]][g[1]]
b = dist[1][g[0]][g[1]]

if (a==-1)&(b==-1):
    print(-1)
elif (a!=-1)&(b!=-1):
    print(min(a,b))
else:
    print(max(a,b))
0