結果

問題 No.2096 Rage With Our Friends
コンテスト
ユーザー chineristAC
提出日時 2022-09-14 21:08:40
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 5,133 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 589 ms
コンパイル使用メモリ 82,224 KB
実行使用メモリ 795,240 KB
最終ジャッジ日時 2024-12-15 12:25:46
合計ジャッジ時間 67,863 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 MLE * 2
other AC * 2 WA * 15 TLE * 9 MLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 


import sys
from collections import *

input = lambda :sys.stdin.readline().rstrip()

"""
考察の内容
01-BFSでよさそうだけどEを持たなきゃいけなさそう？
→前進するとE=0になるので後退→前進の2回分を全探索でよさげ

後退については一番後退せずにすむ点にジャンプするのが得（この後後退する場合もできるだけ前のほうが候補が増える）
前進についてはできるだけ前にだけいった後、好きなだけbackできるように頂点倍化？
"""

H,W = map(int,input().split())
_,sy,_,gy = map(int,input().split())
sy,gy = sy-1,gy-1
S = [input() for i in range(H)]

mini = [[H for j in range(W)] for i in range(H)]
for j in range(W):
    for i in range(H)[::-1]:
        if i!=H-1:
            mini[i][j] = min(mini[i][j],mini[i+1][j])
        if S[i][j]=="#":
            mini[i][j] = i

INF = 10**17

"""
p=1のとき同じ列でback可能
"""
dist = [[[INF for j in range(W)] for i in range(H)] for p in range(2)]
dist[0][H-1][sy] = 0
deq = deque([(0,H-1,sy)])
done = [[[False]*W for i in range(H)] for p in range(2)]
while deq:
    p,h,w = deq.popleft()
    if done[p][h][w]:
        continue
    done[p][h][w] = True

    """
    前進する
    """
    if h!=0:
        if w!=0 and S[h-1][w-1]=="#":
            if dist[0][h-1][w-1] > dist[p][h][w]:
                dist[0][h-1][w-1] = dist[p][h][w]
                deq.appendleft((0,h-1,w-1))
        if w!=W-1 and S[h-1][w+1]=="#":
            if dist[0][h-1][w+1] > dist[p][h][w]:
                dist[0][h-1][w+1] = dist[p][h][w]
                deq.appendleft((0,h-1,w+1))

    """
    後退する
    """
    if w!=0 and mini[h][w-1]!=H:
        np,nh,nw = 0,mini[h][w-1],w-1
        if dist[np][nh][nw] > dist[p][h][w]:
            dist[np][nh][nw] = dist[p][h][w]
            deq.appendleft((np,nh,nw))
        
        E = nh-h
        """
        ここからさらに前進する
        """
        if nw!=0 and mini[max(0,nh-1-E//2)][nw-1]!=H:
            nnp,nnh,nnw = 1,mini[max(0,nh-1-E//2)][nw-1],nw-1
            if dist[nnp][nnh][nnw] > dist[p][h][w]:
                dist[nnp][nnh][nnw] = dist[p][h][w]
                deq.appendleft((nnp,nnh,nnw))
        if nw!=W-1 and mini[max(0,nh-1-E//2)][nw+1]!=H:
            nnp,nnh,nnw = 1,mini[max(0,nh-1-E//2)][nw+1],nw+1
            if dist[nnp][nnh][nnw] > dist[p][h][w]:
                dist[nnp][nnh][nnw] = dist[p][h][w]
                deq.appendleft((nnp,nnh,nnw))
        """
        ブースト
        """
        if nw!=0 and mini[max(0,nh-1-E)][nw-1]!=H:
            nnp,nnh,nnw = 1,mini[max(0,nh-1-E)][nw-1],nw-1
            if dist[nnp][nnh][nnw] > dist[p][h][w] + 1:
                dist[nnp][nnh][nnw] = dist[p][h][w] + 1
                deq.append((nnp,nnh,nnw))
        if nw!=W-1 and mini[max(0,nh-1-E)][nw+1]!=H:
            nnp,nnh,nnw = 1,mini[max(0,nh-1-E)][nw+1],nw+1
            if dist[nnp][nnh][nnw] > dist[p][h][w] + 1:
                dist[nnp][nnh][nnw] = dist[p][h][w] + 1
                deq.append((nnp,nnh,nnw))
        
        

    if w!=W-1 and mini[h][w+1]!=H:
        np,nh,nw = 0,mini[h][w+1],w+1
        if dist[np][nh][nw] > dist[p][h][w]:
            dist[np][nh][nw] = dist[p][h][w]
            deq.appendleft((np,nh,nw))
        
        E = nh-h
        """
        ここからさらに前進する
        """
        if nw!=0 and mini[max(0,nh-1-E//2)][nw-1]!=H:
            nnp,nnh,nnw = 1,mini[max(0,nh-1-E//2)][nw-1],nw-1
            if dist[nnp][nnh][nnw] > dist[p][h][w]:
                dist[nnp][nnh][nnw] = dist[p][h][w]
                deq.appendleft((nnp,nnh,nnw))
        if nw!=W-1 and mini[max(0,nh-1-E//2)][nw+1]!=H:
            nnp,nnh,nnw = 1,mini[max(0,nh-1-E//2)][nw+1],nw+1
            if dist[nnp][nnh][nnw] > dist[p][h][w]:
                dist[nnp][nnh][nnw] = dist[p][h][w]
                deq.appendleft((nnp,nnh,nnw))
        """
        ブースト
        """
        if nw!=0 and mini[max(0,nh-1-E)][nw-1]!=H:
            nnp,nnh,nnw = 1,mini[max(0,nh-1-E)][nw-1],nw-1
            if dist[nnp][nnh][nnw] > dist[p][h][w] + 1:
                dist[nnp][nnh][nnw] = dist[p][h][w] + 1
                deq.append((nnp,nnh,nnw))
        
        #if (nh,nw)==(3,4) and h==0:
            #print("OKOKO")
            #print(mini[max(0,nh-1-E)][nw+1],nw+1)
        if nw!=W-1 and mini[max(0,nh-1-E)][nw+1]!=H:
            nnp,nnh,nnw = 1,mini[max(0,nh-1-E)][nw+1],nw+1
            if dist[nnp][nnh][nnw] > dist[p][h][w] + 1:
                dist[nnp][nnh][nnw] = dist[p][h][w] + 1
                deq.append((nnp,nnh,nnw))
    
    """
    p=1の場合、後退する or p=0に移行する
    """
    if p==1:
        if dist[0][h][w] > dist[p][h][w]:
            dist[0][h][w] = dist[p][h][w]
            deq.appendleft((0,h,w))
        
        if h!=H-1 and mini[h+1][w]!=H:
            np,nh,nw = 1,mini[h+1][w],w
            if dist[np][nh][nw] > dist[p][h][w]:
                dist[np][nh][nw] = dist[p][h][w]
                deq.appendleft((np,nh,nw))

if dist[0][0][gy] == INF:
    print(-1)
else:
    print(dist[0][0][gy])
0