给定一个二维的矩阵,包含 ‘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’。如果两个元素在水平或垂直方向相邻,则称它们是“相连”的。
解析:使用dfs搜索,从边界的O开始搜索,将与之相邻的O全部修改为#
然后再次遍历,将所有的O改为X,将所有的#改为O
class Solution {
public:
void solve(vector<vector<char>>& board) {
int n = board.size();
if(n==0) return;
int m = board[0].size();
if(m==0) return;
for(int i=0;i<m;++i){
if(board[0][i]=='O') dfs(board,0,i);
if(board[n-1][i]=='O') dfs(board,n-1,i);
}
for(int j=0;j<n;++j){
if(board[j][0]=='O') dfs(board,j,0);
if(board[j][m-1]=='O') dfs(board,j,m-1);
}
for(int i=0;i<n;++i){
for(int j=0;j<m;++j){
if(board[i][j]=='O') board[i][j]='X';
else if(board[i][j]=='#') board[i][j]='O';
}
}
}
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') return;
board[i][j]='#';
dfs(board,i+1,j);
dfs(board,i-1,j);
dfs(board,i,j+1);
dfs(board,i,j-1);
}
};