結果

問題 No.340 雪の足跡
ユーザー rpy3cpp
提出日時 2016-01-30 10:52:17
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 2,439 bytes
コンパイル時間 125 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 97,180 KB
最終ジャッジ日時 2024-09-21 19:07:38
合計ジャッジ時間 19,104 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 24 TLE * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

def solve():
    W, H, hsWH, hsHW = read_data()
    build_edges(hsWH, hsHW)
    return wfs(W, H, hsWH, hsHW)


def read_data():
    W, H, N = map(int, input().split())
    hsWH = [[0] * H for w in range(W)]  # vertical lines
    hsHW = [[0] * W for h in range(H)]  # horizontal lines
    for n in range(N):
        process_move(hsWH, hsHW, W, H)
    return W, H, hsWH, hsHW


def process_move(hsWH, hsHW, W, H):
    Mnotused = input()
    Bs = list(map(int, input().split()))
    h0, w0 = divmod(Bs[0], W)
    for b in Bs:
        h1, w1 = divmod(b, W)
        if h0 == h1:  # horizontal move
            if w0 > w1:
                hsHW[h0][w1] += 1
                hsHW[h0][w0] -= 1
            else:
                hsHW[h0][w0] += 1
                hsHW[h0][w1] -= 1
            w0 = w1
        else:        # vertical move
            if h0 > h1:
                hsWH[w0][h1] += 1
                hsWH[w0][h0] -= 1
            else:
                hsWH[w0][h0] += 1
                hsWH[w0][h1] -= 1
            h0 = h1


def build_edges(hsWH, hsHW):
    cum_sum(hsWH)
    cum_sum(hsHW)

def cum_sum(mat):
    for row in mat:
        cum = 0
        for i, val in enumerate(row):
            cum += val
            row[i] = cum

def wfs(W, H, hsWH, hsHW):
    '''
    (0, 0) から (W-1, H-1) までの最短距離を求める。
    hsWH[w][h]: cell(w, h)　と cell(w, h+1) に辺があるか否か
    hsHW[h][w]: cell(w, h)　と cell(w+1, h) に辺があるか否か
    '''
    start = 0
    goal = H * W - 1
    unvisited = [True] * (H * W)
    frontiers = [start]
    cost = 0
    while frontiers:
        cost += 1
        new_frontiers = []
        for f in frontiers:
            if f == goal:
                return cost - 1
            h, w = divmod(f, W)
            if hsWH[w][h] and unvisited[f + W]:
                unvisited[f + W] = False
                new_frontiers.append(f + W)
            if hsWH[w][h-1] and unvisited[f - W]:
                unvisited[f - W] = False
                new_frontiers.append(f - W)
            if hsHW[h][w] and unvisited[f + 1]:
                unvisited[f + 1] = False
                new_frontiers.append(f + 1)
            if hsHW[h][w-1] and unvisited[f - 1]:
                unvisited[f - 1] = False
                new_frontiers.append(f - 1)
        frontiers = new_frontiers
    return -1

ans = solve()
if ans == -1:
    print("Odekakedekinai..")
else:
    print(ans)
0