結果

問題 No.697 池の数はいくつか
ユーザー rin204
提出日時 2022-03-23 16:12:19
言語 PyPy3
(7.3.17)
結果
MLE  
実行時間 -
コード長 782 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 281 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 314,124 KB
最終ジャッジ日時 2024-10-11 19:10:28
合計ジャッジ時間 16,046 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30 MLE * 2
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

h, w = map(int, input().split())
A = []
for _ in range(h):
    A += list(map(int, input().split()))
directions = ((0, 1), (1, 0), (0, -1), (-1, 0))

ans = 0
for i in range(h):
    for j in range(w):
        if A[i * w + j] == 0:
            continue
        A[i * w + j] = 0
        ans += 1
        stack = [(i * w + j)]
        while stack:
            tmp = stack.pop()
            ii = tmp // w
            jj = tmp - ii * w
            for di, dj in directions:
                ni = ii + di
                nj = jj + dj
                if ni == -1 or ni == h or nj == -1 or nj == w:
                    continue
                if A[ni * w + nj] == 0:
                    continue
                A[ni * w + nj] = 0
                stack.append(ni * w + nj)
        
print(ans)
0