結果

問題 No.340 雪の足跡
コンテスト
ユーザー mkawa2
提出日時 2020-04-29 01:32:53
言語 Python3
(3.14.2 + numpy 2.4.0 + scipy 1.16.3)
結果
WA  
実行時間 -
コード長 2,357 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 168 ms
コンパイル使用メモリ 13,184 KB
実行使用メモリ 102,912 KB
最終ジャッジ日時 2024-11-26 08:56:39
合計ジャッジ時間 23,891 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 21 WA * 1 TLE * 10
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

from itertools import *
from bisect import *
from collections import *
from heapq import *
import sys

sys.setrecursionlimit(10 ** 6)

def II(): return int(sys.stdin.readline())
def MI(): return map(int, sys.stdin.readline().split())
def LI(): return list(map(int, sys.stdin.readline().split()))
def SI(): return sys.stdin.readline()[:-1]
def LLI(rows_number): return [LI() for _ in range(rows_number)]
def LLI1(rows_number): return [LI1() for _ in range(rows_number)]
int1 = lambda x: int(x) - 1
def MI1(): return map(int1, sys.stdin.readline().split())
def LI1(): return list(map(int1, sys.stdin.readline().split()))
p2D = lambda x: print(*x, sep="\n")
dij = [(1, 0), (0, 1), (-1, 0), (0, -1)]

def main():
    w,h,n=MI()
    # lrは左右移動、udは上下移動。いもす法の要領で移動可能な区間を記録していく
    lr=[0]*(w*h+1)
    ud=[0]*(w*h+1)
    for _ in range(n):
        SI()
        bb=LI()
        # 区間の始点に+1、終点に-1をしていく
        for b0,b1 in zip(bb,bb[1:]):
            if b0>b1:b0,b1=b1,b0
            # 左右移動の場合
            if b0//w==b1//w:
                lr[b0+1]+=1
                lr[b1+1]-=1
            # 上下移動の場合
            else:
                j0=b0%w*h+b0//w
                j1=b1%w*h+b1//w
                ud[j0+1]+=1
                ud[j1+1]-=1
    # 累積和をとって移動可能な区間を求める
    for i in range(w*h):lr[i+1]+=lr[i]
    for j in range(w*h):ud[j+1]+=ud[j]
    #print(lr)
    #print(ud)

    q=deque()
    q.append((0,0))
    first=[True]*w*h

    def finish(d):
        print(d+1)
        exit()

    while q:
        d,i=q.popleft()
        j=i%w*h+i//w
        # 左への移動
        if lr[i] and first[i-1]:
            if i-1==h*w-1:finish(d)
            first[i-1]=False
            q.append((d+1,i-1))
        # 右への移動
        if lr[i+1] and first[i+1]:
            if i+1==h*w-1:finish(d)
            first[i+1]=False
            q.append((d+1,i+1))
        # 上への移動
        if ud[j+1] and first[i+w]:
            if i+w==h*w-1:finish(d)
            first[i+w]=False
            q.append((d+1,i+w))
        # 下への移動
        if ud[j] and first[i-w]:
            if i-w==h*w-1:finish(d)
            first[i-w]=False
            q.append((d+1,i-w))

    print("Odekakedekinai..")

main()
0