結果

問題 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()
0