【Leetcode 广搜、深搜】扫雷游戏（529）

题目

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

给定一个代表游戏板的二维字符矩阵。 '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,50]。
点击的位置只能是未被挖出的方块 ('M' 或者 'E')，这也意味着面板至少包含一个可点击的方块。
输入面板不会是游戏结束的状态（即有地雷已被挖出）。
简单起见，未提及的规则在这个问题中可被忽略。例如，当游戏结束时你不需要挖出所有地雷，考虑所有你可能赢得游戏或标记方块的情况。

解答

理解题意并不复杂，典型的搜索题，深搜广搜都可以，代码为广搜：

from queue import Queue
from pprint import pprint


# BFS
# Time: O(MN), Space: O(MN), M N表示输入数组行列数
class Solution:
    def get_cnt(self, next, x, y, rows, columns, board):
        '''获取周围炸弹个数'''
        cnt = 0
        for i in range(8):
            tx = x + next[i][0]
            ty = y + next[i][1]
            if 0 <= tx < rows and 0 <= ty < columns and board[tx][ty] == 'M':
                cnt += 1
        return cnt

    def updateBoard(self, board, click):
        if not board:
            return board
        if board[click[0]][click[1]] == 'M':
            board[click[0]][click[1]] = 'X'
            return board
        rows, columns = len(board), len(board[0])
        next = [
            [0, 1],
            [1, 0],
            [0, -1],
            [-1, 0],
            [1, 1],  # 右下
            [1, -1],
            [-1, -1],
            [-1, 1]
        ]

        # 点击的位置周围有炸弹，修改为炸弹数量，结束
        cnt = self.get_cnt(next, click[0], click[1], rows, columns, board)
        if cnt:
            board[click[0]][click[1]] = str(cnt)
            return board

        q = Queue()
        q.put(click)
        board[click[0]][click[1]] = 'B'
        while not q.empty():
            x, y = q.get()
            for i in range(8):
                tx = x + next[i][0]
                ty = y + next[i][1]
                if 0 <= tx < rows and 0 <= ty < columns and board[tx][ty] == 'E':
                    cnt = self.get_cnt(next, tx, ty, rows, columns, board)  # 周围炸弹个数
                    if not cnt:
                        board[tx][ty] = 'B'
                        q.put([tx, ty])
                    else:
                        board[tx][ty] = str(cnt)  # 周围有炸弹，修改为炸弹数量，不计入队列
        return board


# [["B", "B", "B", "B", "B", "B", "1", "1"],
#  ["B", "1", "1", "1", "B", "B", "1", "M"],
#  ["1", "2", "M", "1", "B", "B", "1", "1"],
#  ["M", "2", "1", "1", "B", "B", "B", "B"],
#  ["1", "1", "B", "B", "B", "B", "B", "B"],
#  ["B", "B", "B", "B", "B", "B", "B", "B"],
#  ["B", "1", "2", "2", "1", "B", "B", "B"],
#  ["B", "1", "M", "M", "1", "B", "B", "B"]]


s = Solution()
ans = s.updateBoard([["B", "B", "B", "B", "B", "B", "1", "E"],
                     ["B", "1", "1", "1", "B", "B", "1", "M"],
                     ["1", "2", "M", "1", "B", "B", "1", "1"],
                     ["M", "2", "1", "1", "B", "B", "B", "B"],
                     ["1", "1", "B", "B", "B", "B", "B", "B"],
                     ["B", "B", "B", "B", "B", "B", "B", "B"],
                     ["B", "1", "2", "2", "1", "B", "B", "B"],
                     ["B", "1", "M", "M", "1", "B", "B", "B"]],
                    [0, 7])
# ans = s.updateBoard([['E', 'E', 'E', 'E', 'E'],
#                      ['E', 'E', 'M', 'E', 'E'],
#                      ['E', 'E', 'E', 'E', 'E'],
#                      ['E', 'E', 'E', 'E', 'E']], [3, 0])
pprint(ans)