結果

問題 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
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

import sys
input = sys.stdin.readline
from collections import defaultdict
from heapq import heappop,heappush,heapify
h,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]=0
que=[[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%2
for di,dj in [(1,0),(0,1),(-1,0),(0,-1)]:
ni,nj=i+di,j+dj
if 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()
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0