题目

难度：★★★☆☆
类型：二维数组
方法：深度优先搜索

力扣链接请移步本题传送门
更多力扣中等题的解决方案请移步力扣中等题目录

让我们一起来玩扫雷游戏！

给定一个代表游戏板的二维字符矩阵。 'M' 代表一个未挖出的地雷，'E' 代表一个未挖出的空方块，'B' 代表没有相邻（上，下，左，右，和所有4个对角线）地雷的已挖出的空白方块，数字（'1' 到 '8'）表示有多少地雷与这块已挖出的方块相邻，'X' 则表示一个已挖出的地雷。

现在给出在所有未挖出的方块中（'M'或者'E'）的下一个点击位置（行和列索引），根据以下规则，返回相应位置被点击后对应的面板：

如果一个地雷（'M'）被挖出，游戏就结束了- 把它改为 'X'。
如果一个没有相邻地雷的空方块（'E'）被挖出，修改它为（'B'），并且所有和其相邻的未挖出方块都应该被递归地揭露。
如果一个至少与一个地雷相邻的空方块（'E'）被挖出，修改它为数字（'1'到'8'），表示相邻地雷的数量。
如果在此次点击中，若无更多方块可被揭露，则返回面板。

示例

示例 1：

输入:

[['E', 'E', 'E', 'E', 'E'],
['E', 'E', 'M', 'E', 'E'],
['E', 'E', 'E', 'E', 'E'],
['E', 'E', 'E', 'E', 'E']]

Click : [3,0]

输出:

[['B', '1', 'E', '1', 'B'],
['B', '1', 'M', '1', 'B'],
['B', '1', '1', '1', 'B'],
['B', 'B', 'B', 'B', 'B']]

示例 2：

输入:

[['B', '1', 'E', '1', 'B'],
['B', '1', 'M', '1', 'B'],
['B', '1', '1', '1', 'B'],
['B', 'B', 'B', 'B', 'B']]

Click : [1,2]

输出:

[['B', '1', 'E', '1', 'B'],
['B', '1', 'X', '1', 'B'],
['B', '1', '1', '1', 'B'],
['B', 'B', 'B', 'B', 'B']]

解答

方法1：深度优先搜索

class Solution:
    def updateBoard(self, board, click):

        x, y = click
        if board[x][y] == "M":
            board[x][y] = "X"
            return board

        row, col = len(board), len(board[0])

        def get_surroundings(x, y):
            dx_dy = [(-1, -1), (-1, 0), (-1, 1), (0, -1), (0, 1), (1, -1), (1, 0), (1, 1)]
            surroundings = [(x+dx, y+dy) for dx, dy in dx_dy]
            return list(filter(lambda loc: 0 <= loc[0] < row and 0 <= loc[1] < col, surroundings))

        def dfs(x, y):
            num = sum([1 if board[x][y] == 'M' else 0 for x, y in get_surroundings(x, y)])
            if num > 0:
                board[x][y] = str(num)
            else:
                board[x][y] = "B"
                for x_, y_ in get_surroundings(x, y):
                    if board[x_][y_] == "E":
                        dfs(x_, y_)

        dfs(x, y)
        return board

方法2：宽度优先搜索

class Solution:
    def updateBoard(self, board, click):
        x, y = click
        row, col = len(board), len(board[0])
        if board[x][y] == "M":
            board[x][y] = "X"
            return board

        # 计算空白快周围的***
        def get_surroundings(x, y):
            dx_dy = [(-1, -1), (-1, 0), (-1, 1), (0, -1), (0, 1), (1, -1), (1, 0), (1, 1)]
            surroundings = [(x+dx, y+dy) for dx, dy in dx_dy]
            return list(filter(lambda loc: 0 <= loc[0] < row and 0 <= loc[1] < col, surroundings))

        def bfs(x, y):
            queue = [[x, y]]
            while queue:
                x, y = queue.pop()
                num = sum([1 if board[x][y] == 'M' else 0 for x, y in get_surroundings(x, y)])
                if num > 0:
                    board[x][y] = str(num)
                    continue
                board[x][y] = "B"
                for x_, y_ in get_surroundings(x, y):
                    if board[x_][y_] == "E":
                        queue = [[x_, y_]] + queue
                        board[x_][y_] = "B"

        bfs(x, y)
        return board

如有疑问或建议，欢迎评论区留言~

有关更多力扣中等题的python解决方案，请移步力扣中等题解析