結果
問題 |
No.2731 Two Colors
|
ユーザー |
|
提出日時 | 2024-04-19 22:23:36 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,366 ms / 3,000 ms |
コード長 | 831 bytes |
コンパイル時間 | 329 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 124,420 KB |
最終ジャッジ日時 | 2024-10-11 15:52:22 |
合計ジャッジ時間 | 16,125 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 33 |
ソースコード
import heapq D=[[1,0],[0,1],[-1,0],[0,-1]] H,W=map(int,input().split()) A=[] for i in range(H): A.append([int(_) for _ in input().split()]) C=[[None for j in range(W)] for i in range(H)] Q=[[(A[H-1][W-1],(H-1,W-1))],[(A[0][0],(0,0))]] heapq.heapify(Q[1]) heapq.heapify(Q[0]) step = -2 while True: step += 1 k = step%2 while True: _, p = heapq.heappop(Q[k]) Y, X = p if C[Y][X] is None: break C[Y][X]=k f = False for d in D: y = Y + d[0] x = X + d[1] if 0<=y<H and 0<=x<W: if C[y][x]==1-k: f = True break if C[y][x] is None: heapq.heappush(Q[k], (A[y][x],(y,x))) if f: break print(step) ''' for c in C: print(*[' ' if _ is None else _ for _ in c]) '''