結果
問題 | No.2731 Two Colors |
ユーザー |
![]() |
提出日時 | 2024-09-24 20:55:42 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 888 ms / 3,000 ms |
コード長 | 982 bytes |
コンパイル時間 | 259 ms |
コンパイル使用メモリ | 82,188 KB |
実行使用メモリ | 267,956 KB |
最終ジャッジ日時 | 2024-09-24 20:56:00 |
合計ジャッジ時間 | 13,340 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 33 |
ソースコード
import sysinput = sys.stdin.readlinefrom collections import defaultdictfrom heapq import heappop,heappush,heapifyh,w=map(int,input().split())A=[list(map(int,input().split())) for _ in range(h)]pos=defaultdict(tuple)for i in range(h):for j in range(w):pos[A[i][j]]=(i,j)color=[[-1]*w for _ in range(h)]color[0][0]=1;color[h-1][w-1]=0que=[[A[h-1][w-2],A[h-2][w-1]],[A[0][1],A[1][0]]]heapify(que[0]);heapify(que[1])s=[set(),set()]for i in range(2):for a in que[i]:s[i].add(a)for turn in range(1,h*w+1):a=heappop(que[turn%2])i,j=pos[a]color[i][j]=turn%2for di,dj in [(1,0),(0,1),(-1,0),(0,-1)]:ni,nj=i+di,j+djif 0<=ni<h and 0<=nj<w and color[ni][nj]==-1 and A[ni][nj] not in s[turn%2]:s[turn%2].add(A[ni][nj])heappush(que[turn%2],(A[ni][nj]))if 0<=ni<h and 0<=nj<w and color[ni][nj]==(1-turn%2):print(turn)exit()