結果
問題 |
No.2731 Two Colors
|
ユーザー |
|
提出日時 | 2024-04-19 22:22:55 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
TLE
|
実行時間 | - |
コード長 | 831 bytes |
コンパイル時間 | 304 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 73,664 KB |
最終ジャッジ日時 | 2024-10-11 15:51:46 |
合計ジャッジ時間 | 9,523 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | -- * 3 |
other | TLE * 1 -- * 32 |
ソースコード
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]) '''