leetcode_130_被围绕的区域

被围绕的区域

描述

中等

给定一个二维的矩阵，包含 'X' 和 'O'（字母 O）。

找到所有被 'X' 围绕的区域，并将这些区域里所有的 'O' 用 'X' 填充。

示例:

X X X X
X O O X
X X O X
X O X X

运行你的函数后，矩阵变为：

X X X X
X X X X
X X X X
X O X X

解释:

被围绕的区间不会存在于边界上，换句话说，任何边界上的 'O' 都不会被填充为 'X'。 任何不在边界上，或不与边界上的 'O' 相连的 'O' 最终都会被填充为 'X'。如果两个元素在水平或垂直方向相邻，则称它们是“相连”的。

解题

边界上的'O'和取其相连的'O'不会被填充为'X'，其余都会被填充成'X'

那么就从边界上出发，使用DFS找出与边界上的'O'相连的'O'，标记为'A'

然后逐个扫描，如果是'O'，那就是不与边界相接的'O'，填充为'X'

如果是'A'，那就是原来边界上的'O'或与边界相连的'O'，恢复为'O'

class Solution:
    def solve(self, board: List[List[str]]) -> None:
        """
        Do not return anything, modify board in-place instead.
        """
        if not board:
            return []

        n, m = len(board), len(board[0])

        def dfs(x, y):
            if x < 0 or y < 0 or y >= m or x >= n or board[x][y] != 'O':
                return

            board[x][y] = 'A'
            dfs(x + 1, y)
            dfs(x - 1, y)
            dfs(x, y + 1)
            dfs(x, y - 1)

        # 扫描边界，使用DFS
        for i in range(n):
            dfs(i, 0)
            dfs(i, m - 1)
        for i in range(1, m - 1):
            dfs(0, i)
            dfs(n - 1, i)

        for i in range(n):
            for j in range(m):
                if board[i][j] == 'A':
                    board[i][j] = 'O'
                elif board[i][j] == 'O':
                    board[i][j] = 'X'