给定一个二维的矩阵,包含 ‘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’,则把’O’换成’#’;接着dfs这个位置上下左右是否有’O’如果有继续换成’#’;最后把’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 None
row = len(board)
col = len(board[0])
if row < 3 or col < 3:
return None
def dfs(i,j):
if i<0 or i>= row or j<0 or j>= col or board[i][j]!='O':
return
if board[i][j] == 'O':
board[i][j] = '#'
dfs(i-1,j)
dfs(i+1,j)
dfs(i,j-1)
dfs(i,j+1)
for i in range(row):
dfs(i,0)
dfs(i,col-1)
for j in range(col):
dfs(0,j)
dfs(row-1,j)
for i in range(row):
for j in range(col):
if(board[i][j]) == 'O':
board[i][j] = 'X'
if(board[i][j]== '#'):
board[i][j] = 'O'