13 - Hard - 被围绕的区域

2018-06-28  本文已影响0人  1f872d1e3817

给定一个二维的矩阵,包含 '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再把该点上下左右的点加入遍历队列

Phase 1: "Save" every O-region touching the border, changing its cells to 'S'.
Phase 2: Change every 'S' on the board to 'O' and everything else to 'X'.

class Solution(object):
    def solve(self, board):
        """
        :type board: List[List[str]]
        :rtype: void Do not return anything, modify board in-place instead.
        """
        if not any(board): 
            return
        m, n = len(board), len(board[0])
        save = [ij for k in range(max(m, n)) for ij in ((0, k), (m-1, k), (k, 0), (k, n-1))]
        while save:
            i, j = save.pop()
            if 0 <= i < m and 0 <= j < n and board[i][j] == 'O':
                board[i][j] = 'S'
                save += (i, j-1), (i, j+1), (i-1, j), (i+1, j)

        board[:] = [['XO'[c == 'S'] for c in row] for row in board]

更详细的BFS

class Solution(object):
    def solve(self, board):
        """
        :type board: List[List[str]]
        :rtype: void Do not return anything, modify board in-place instead.
        """

        def BFS(matrix, index_i, index_j):
            stack = [[index_i, index_j]]
            steps = [[-1, 0], [1, 0], [0, -1], [0, 1]]

            while stack:
                tmp_point = stack.pop()
                matrix[tmp_point[0]][tmp_point[1]] = '-1'
                for step in steps:
                    right_i = tmp_point[0] + step[0]
                    right_j = tmp_point[1] + step[1]
                    if right_i >= 0 and right_i < len(matrix) and right_j >= 0 and right_j < len(matrix[0]) \
                            and matrix[right_i][right_j] == 'O':
                        stack.append([right_i, right_j])

            return matrix

        if board != []:
            for i in range(0, len(board)):
                if board[i][0] == 'O':
                    board = BFS(board, i, 0)
                if board[i][len(board[0]) - 1] == 'O':
                    board = BFS(board, i, len(board[0]) - 1)

            for j in range(0, len(board[0])):
                if board[0][j] == 'O':
                    board = BFS(board, 0, j)
                if board[len(board) - 1][j] == 'O':
                    board = BFS(board, len(board) - 1, j)

            for i in range(0, len(board)):
                for j in range(0, len(board[0])):
                    if board[i][j] == 'O':
                        board[i][j] = 'X'
                    if board[i][j] == '-1':
                        board[i][j] = 'O'
上一篇下一篇

猜你喜欢

热点阅读