結果

問題 No.697 池の数はいくつか
ユーザー ryusuke
提出日時 2022-02-04 21:04:58
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,087 ms / 6,000 ms
コード長 1,186 bytes
コンパイル時間 313 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 281,548 KB
最終ジャッジ日時 2024-11-08 09:02:19
合計ジャッジ時間 15,628 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import deque

h, w = map(int, input().split())
a = [list(map(int, input().split())) for _ in range(h)]

# dist[i][j] := INF -> 池でまだ見てない, -1 -> 地面で通れない
dist = [[-1] * w for _ in range(h)]
for i in range(h):
    for j in range(w):
        if a[i][j]:
            dist[i][j] = 0

cnt = 1 # 今見ている池の番号
q = deque()
for i in range(h):
    for j in range(w):
        if a[i][j]:
            q.append((i, j))
            dist[i][j] = cnt
            break
    break

d = ((1, 0), (-1, 0), (0, 1), (0, -1))
for i in range(h):
    for j in range(w):
        if not (dist[i][j] == 0 or dist[i][j] == cnt): continue
        if dist[i][j] == 0:
            q.append((i, j))
        while q:
            vy, vx = q.popleft()
            dist[vy][vx] = cnt
            for dy, dx in d:
                y = vy + dy
                x = vx + dx
                if not (0 <= x < w and 0 <= y < h): continue
                if dist[y][x] != 0: continue
                dist[y][x] = dist[vy][vx]
                q.append((y, x))
        cnt += 1

ans = 0
for i in range(h):
    for j in range(w):
        ans = max(ans, dist[i][j])

print(ans)
0