結果

問題 No.2731 Two Colors
ユーザー なえしらなえしら
提出日時 2024-08-13 23:49:26
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 770 ms / 3,000 ms
コード長 652 bytes
コンパイル時間 397 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 264,196 KB
最終ジャッジ日時 2024-08-13 23:49:39
合計ジャッジ時間 12,417 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 770 ms
263,948 KB
testcase_01 AC 731 ms
264,196 KB
testcase_02 AC 757 ms
263,676 KB
testcase_03 AC 87 ms
76,956 KB
testcase_04 AC 113 ms
83,840 KB
testcase_05 AC 107 ms
80,576 KB
testcase_06 AC 165 ms
99,072 KB
testcase_07 AC 163 ms
106,624 KB
testcase_08 AC 101 ms
79,104 KB
testcase_09 AC 351 ms
156,400 KB
testcase_10 AC 93 ms
77,156 KB
testcase_11 AC 424 ms
145,180 KB
testcase_12 AC 368 ms
161,100 KB
testcase_13 AC 381 ms
157,516 KB
testcase_14 AC 216 ms
122,636 KB
testcase_15 AC 101 ms
78,392 KB
testcase_16 AC 225 ms
115,840 KB
testcase_17 AC 259 ms
137,160 KB
testcase_18 AC 119 ms
83,712 KB
testcase_19 AC 233 ms
115,456 KB
testcase_20 AC 259 ms
130,016 KB
testcase_21 AC 131 ms
84,552 KB
testcase_22 AC 396 ms
188,660 KB
testcase_23 AC 171 ms
99,840 KB
testcase_24 AC 133 ms
88,576 KB
testcase_25 AC 95 ms
77,184 KB
testcase_26 AC 315 ms
174,848 KB
testcase_27 AC 386 ms
165,356 KB
testcase_28 AC 289 ms
135,424 KB
testcase_29 AC 157 ms
95,232 KB
testcase_30 AC 203 ms
116,224 KB
testcase_31 AC 171 ms
96,896 KB
testcase_32 AC 470 ms
174,888 KB
testcase_33 AC 38 ms
52,864 KB
testcase_34 AC 39 ms
52,864 KB
testcase_35 AC 38 ms
52,992 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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)]

IJ = {A[i][j]: (i, j) for i in range(H) for j in range(W)}

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

for t in range(-1, H*W-1):
  i, j = IJ[heappop(que[t%2])]
  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 or color[ni][nj] == t%2-3:
      color[ni][nj] -= t%2+1
      heappush(que[t%2], A[ni][nj])
    elif color[ni][nj] == 1-t%2: break
  else: continue

  print(t); break
0