Leetcode练习(Python):深度优先搜索类:第130题:被围绕的区域:给定一个二维的矩阵,包含 'X' 和 'O',字母 O。 找到所有被 'X' 围绕的区域,并将这些区域里所有的 'O' 用 'X' 填充。

题目:

被围绕的区域:给定一个二维的矩阵,包含 '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'变为'X',将’$'变为‘O’。

程序:

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

        if len(board) < 3 or len(board[0]) < 3: 
            return
      
        for i in range(len(board)):
            self.auxiliary(board, i, 0)
            self.auxiliary(board, i, len(board[0]) - 1)
        
        for i in range(len(board[0])):
            self.auxiliary(board, 0, i)
            self.auxiliary(board, len(board) - 1, i)

        for i in range(len(board)):
            for j in range(len(board[0])):
                if board[i][j] == 'O':
                    board[i][j] = 'X'
                if board[i][j] == '$':
                    board[i][j] = 'O'

    def auxiliary(self, board: List[List[str]], index1: int, index2: int):
        if index1 < 0 or index2 < 0 or index1 >= len(board) or index2 >= len(board[0]) or board[index1][index2] != 'O':
            return
        board[index1][index2] = '$'
        self.auxiliary(board, index1 - 1, index2)
        self.auxiliary(board, index1 + 1, index2)
        self.auxiliary(board, index1, index2 - 1)
        self.auxiliary(board, index1, index2 + 1)