結果

問題 No.2731 Two Colors
ユーザー るこーそーるこーそー
提出日時 2024-09-24 20:55:18
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 945 bytes
コンパイル時間 180 ms
コンパイル使用メモリ 82,508 KB
実行使用メモリ 229,844 KB
最終ジャッジ日時 2024-09-24 20:55:34
合計ジャッジ時間 15,174 ms
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 AC 42 ms
54,436 KB
testcase_34 AC 39 ms
55,044 KB
testcase_35 AC 40 ms
54,916 KB
権限があれば一括ダウンロードができます

ソースコード

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]:
            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