結果

問題 No.697 池の数はいくつか
ユーザー sjiki
提出日時 2024-08-05 18:43:49
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,075 bytes
コンパイル時間 316 ms
コンパイル使用メモリ 81,848 KB
実行使用メモリ 54,548 KB
最終ジャッジ日時 2024-08-05 18:43:54
合計ジャッジ時間 5,243 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 3
other WA * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

def count_ponds(field):
    def dfs(x, y):
        stack = [(x, y)]
        while stack:
            cx, cy = stack.pop()
            for dx, dy in [(-1, 0), (1, 0), (0, -1), (0, 1)]:
                nx, ny = cx + dx, cy + dy
                if 0 <= nx < n and 0 <= ny < m and field[nx][ny] == 'W' and not visited[nx][ny]:
                    visited[nx][ny] = True
                    stack.append((nx, ny))

    n = len(field)
    m = len(field[0])
    visited = [[False] * m for _ in range(n)]
    pond_count = 0

    for i in range(n):
        for j in range(m):
            if field[i][j] == 'W' and not visited[i][j]:
                visited[i][j] = True
                dfs(i, j)
                pond_count += 1

    return pond_count

# Example usage
if __name__ == "__main__":
    field = [
        "W........WW.",
        ".WWW.....WWW",
        "....WW...WW.",
        ".........WW.",
        ".........W..",
        "..W......W..",
        ".W.W.....WW.",
        "W.W.W.....W.",
        ".W.W......W.",
        "..W.......W."
    ]
    print(count_ponds(field))
0