【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)