给定一个二维的矩阵,包含 '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出发,用DFS把所有相邻的的O染成P。
此时没有被染色的点就是被围绕的区域。
再扫描一次,把所有的P变成O,把所有的O变成X即可。
class Solution(object):
def solve(self, board):
"""
:type board: List[List[str]]
:rtype: None Do not return anything, modify board in-place instead.
"""
if not board or not board[0]:
return board
m, n = len(board), len(board[0])
dx = [1, -1, 0, 0]
dy = [0, 0, 1, -1]
def dfs(x, y):
for k in range(4):
xx = x + dx[k]
yy = y + dy[k]
if 0 <= xx < m and 0 <= y < n: #当前坐标有效
if board[xx][yy] == "O":
board[xx][yy] = "P" #给走过的点染色
dfs(xx, yy)
#--------以下在从四周往里找相邻的所有点
i = 0
for j in range(n): #上边
if board[i][j] == "O":
board[i][j] = "P"
dfs(i, j)
i = m - 1
for j in range(n): #下边
if board[i][j] == "O":
board[i][j] = "P"
dfs(i, j)
j = 0
for i in range(m): #左边
if board[i][j] == "O":
board[i][j] = "P"
dfs(i, j)
j = n - 1
for i in range(m): #右边
if board[i][j] == "O":
board[i][j] = "P"
dfs(i, j)
#--------以上在从四周往里找相邻的所有点
# print board
res = 0
for i in range(m):
for j in range(n):
if board[i][j] == "P": #染色回来
board[i][j] = "O"
elif board[i][j] == "O":
board[i][j] = "X"
return board
下面的写于2019年7月5日22:49:39
class Solution(object):
def solve(self, board):
"""
:type board: List[List[str]]
:rtype: None Do not return anything, modify board in-place instead.
"""
#从四条边的O往里dfs找相邻的O染色成P,到不了的O变成X, P染色回O
if not board or not board[0]:
return board
m, n = len(board), len(board[0])
dx = [1, -1, 0, 0]
dy = [0, 0, 1, -1]
def dfs(x0, y0):
for k in range(4):
x = x0 + dx[k]
y = y0 + dy[k]
if 0 <= x < m and 0 <= y < n and board[x][y] == "O":
board[x][y] = "P"
dfs(x, y)
for i in [0, m - 1]:
for j in range(n):
if board[i][j] == "O":
board[i][j] = "P"
dfs(i, j)
for j in [0, n - 1]:
for i in range(m):
if board[i][j] == "O":
board[i][j] = "P"
dfs(i, j)
for i in range(m):
for j in range(n):
if board[i][j] == "O":
board[i][j] = "X"
elif board[i][j] == "P":
board[i][j] = "O"