题目描述
给定一个二维的矩阵,包含 ‘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’,那么就进行深度优先搜索。标记好这些未被包围的点,最后再进行一次历遍,将除被标记的点全部写为‘X’。
这里我申请了一个flag矩阵。其实可以不用申请,在原矩阵上直接做标记就可以。
class Solution {
public:
void solve(vector<vector<char>>& board) {
if(board.empty()) return;
int row=board.size(), column=board[0].size();
vector<vector<bool>> flag(row,vector<bool>(column,false));
for(int i=0;i<row;++i)
{
if('O'==board[i][0]) dfs(board, flag,i,0);
if('O'==board[i][column-1]) dfs(board, flag, i, column-1);
}
for(int j=0;j<column;++j)
{
if('O'==board[0][j]) dfs(board,flag,0,j);
if('O'==board[row-1][j]) dfs(board,flag,row-1,j);
}
for(int i=0;i<row;++i)
{
for(int j=0;j<column;++j)
{
if(false==flag[i][j]) board[i][j]='X';
}
}
}
void dfs(const vector<vector<char>> &board,vector<vector<bool>> &flag, int i,int j)
{
if('O'==board[i][j] && false==flag[i][j])
{
flag[i][j]=true;
if(i-1>=0) dfs(board, flag, i-1, j);
if(i+1<board.size()) dfs(board, flag, i+1,j);
if(j-1>=0) dfs(board, flag, i,j-1);
if(j+1<board[0].size()) dfs(board, flag, i,j+1);
}
}
};