結果

問題 No.2731 Two Colors
コンテスト
ユーザー lollipop
提出日時 2024-04-19 22:53:18
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 2,549 ms / 3,000 ms
コード長 1,218 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 333 ms
コンパイル使用メモリ 85,248 KB
実行使用メモリ 138,400 KB
最終ジャッジ日時 2026-04-28 00:55:18
合計ジャッジ時間 25,996 ms
ジャッジサーバーID
（参考情報） 		judge3_1 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

def main():  

    dxy = [(1,0), (-1,0), (0,1), (0,-1)]
    from heapq import heapify, heappop, heappush

    H, W = map(int,input().split())
    A = [list(map(int,input().split())) for _ in range(H)]

    visited = [[None for _ in range(W)] for _ in range(H)]
    visited[0][0] = 0
    visited[H-1][W-1] = 1

    hq = [[], []]
    heapify(hq[0]); heapify(hq[1])
    for i, xy in enumerate([[0, 0], [H-1, W-1]]):
        x, y = xy
        for dx, dy in dxy:
            _x = x+dx
            _y = y+dy
            if (not 0 <= _x < H) or (not 0 <= _y < W):
                continue
            heappush(hq[i], [A[_x][_y], _x, _y])


    i = 0
    while True:
        ii = i % 2
        a, x, y = heappop(hq[ii])
        if visited[x][y] != None:
            continue
        visited[x][y] = ii

        for dx, dy in dxy:
            _x = x+dx
            _y = y+dy
            if (not 0 <= _x < H) or (not 0 <= _y < W):
                continue
            if visited[_x][_y] == ii:
                continue
            if visited[_x][_y] == (ii+1) % 2:
                print(i+1)
                return 0
            heappush(hq[ii], [A[_x][_y], _x, _y])
        i += 1

if __name__ == '__main__':  
    exit(main())
0