結果

問題 No.2731 Two Colors
コンテスト
ユーザー なえしら
提出日時 2024-08-13 23:20:20
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 2,155 ms / 3,000 ms
コード長 611 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 348 ms
コンパイル使用メモリ 85,084 KB
実行使用メモリ 135,432 KB
最終ジャッジ日時 2026-04-03 21:06:25
合計ジャッジ時間 18,793 ms
ジャッジサーバーID
（参考情報） 		judge2_0 / judge5_1
純コード判定待ち
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

from heapq import heappop, heappush

DIJ = [(-1, 0), (0, -1), (0, 1), (1, 0)]

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

color = [[-1]*W for _ in range(H)]
que = [[(A[-1][-1], H-1, W-1)], [(A[0][0], 0, 0)]]

for t in range(-1, H*W-1):
  while True:
    a, i, j = heappop(que[t%2])
    if color[i][j] == -1: break
  color[i][j] = t%2
  for ni, nj in DIJ:
    ni += i; nj += j
    if not H > ni >= 0 <= nj < W: continue
    if color[ni][nj] == -1:
      heappush(que[t%2], (A[ni][nj], ni, nj))
    elif color[ni][nj] != t%2: break
  else: continue
  print(t); break
0