DFS之被围绕的区域

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’。如果两个元素在水平或垂直方向相邻，则称它们是“相连”的。

思路
深度优先搜索
由于边界上的’0’以及与边界上的’0’相连的’0’都不会被填充，内部包围的’0’肯定不与边界相连，所以，我们可以从边界进行深度优先搜索，对所有的’0’做标记，那么其余没有被标记的点就全部应该变为’X’．
看代码

class Solution:
    def solve(self, board: List[List[str]]) -> None:
        """
        Do not return anything, modify board in-place instead.
        """
        
        def dfs(board, r, c):
            if 0 <= r < rows and 0 <= c < cols:
                if outside[r][c]:
                    return
                if board[r][c] == 'O': # 如果当前元素为'O'，将其标记，并对其上下左右做下一步的dfs处理
                    outside[r][c] = True
                    dfs(board, r-1, c)
                    dfs(board, r+1, c)
                    dfs(board, r, c-1)
                    dfs(board, r, c+1)
        
        if not board:
            return
        rows = len(board)
        cols = len(board[0])
        outside = [[False] * cols for _ in range(rows)]　# 用来标记边界上及和边界相连的'O'
        for i in range(rows):　# 对第一列和最后一列进行遍历处理，如果元素为'O'，进行dfs遍历处理
            if board[i][0] == 'O':
                dfs(board, i, 0)
            if board[i][cols-1] == 'O':
                dfs(board, i, cols-1)
        for i in range(cols):　# 对第一行和最后一行进行遍历处理，如果元素为'O'，进行dfs遍历处理
            if board[0][i] == 'O':
                dfs(board, 0, i)
            if board[rows-1][i] == 'O':
                dfs(board, rows-1, i)
		# 标记完成之后，对所有未标记的元素置为'X'，包括内部包围的'O'
        for r in range(rows):
            for c in range(cols):
                if not outside[r][c]:
                    board[r][c] = 'X'