题目描述:
给定一个二维的矩阵,包含 ‘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都变成另一个字符,比如1
然后遍历整个矩阵,遇到O变成X,遇到1变成O
代码如下:
class Solution {
public:
void dfs(vector<vector<char>>& board,int i ,int j){
if(i>=0&&j>=0&&i<board.size()&&j<board[0].size()&&board[i][j]=='O'){
board[i][j]='1';
dfs(board,i-1,j);//上
dfs(board,i+1,j);//下
dfs(board,i,j-1);//左
dfs(board,i,j+1);//右
}
}
void solve(vector<vector<char>>& board) {
if(board.size()==0) return;
int row=board.size();
int col=board[0].size();
for(int i=0;i<col;i++){//上下边界
dfs(board,0,i);
dfs(board,row-1,i);
}
for(int i=0;i<row;i++){//左右边界
dfs(board,i,0);
dfs(board,i,col-1);
}
for(int i=0;i<row;i++){
for(int j=0;j<col;j++){
if(board[i][j]=='1'){
board[i][j]='O';
}
else{
board[i][j]='X';
}
}
}
}
};
python:
class Solution:
def solve(self, board: List[List[str]]) -> None:
"""
Do not return anything, modify board in-place instead.
"""
if not board:
return
row=len(board)
col=len(board[0])
for i in range(0,row):
self.dfs(board,i,0)
self.dfs(board,i,col-1)
for j in range(0,col):
self.dfs(board,0,j)
self.dfs(board,row-1,j)
for i in range(0,row):
for j in range(0,col):
if board[i][j]=='*':
board[i][j]='O'
else: board[i][j]='X'
def dfs(self,board,i,j):
if i>=0 and j>=0 and i<len(board) and j<len(board[0]) and board[i][j]=='O':
board[i][j]='*'
self.dfs(board,i+1,j)
self.dfs(board,i-1,j)
self.dfs(board,i,j+1)
self.dfs(board,i,j-1)