题目描述
给你一个 m x n 的矩阵 board ,由若干字符 ‘X’ 和 ‘O’ ,找到所有被 ‘X’ 围绕的区域,并将这些区域里所有的 ‘O’ 用 ‘X’ 填充。
示例 1:
输入:board = [[“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’。如果两个元素在水平或垂直方向相邻,则称它们是“相连”的。
示例 2:
输入:board = [[“X”]]
输出:[[“X”]]
洪水灌溉算法是深度优先遍历,即像洪水一样扩散,如果可以到达的地方那么就属于同一类。本题目已知边界上的 O 不会被填充为 X ,那么可以从此出发来进行洪水灌溉,找到这样的无法被围绕的区域做标记,最后将这些标记位置换为 O,其余位置换为 X。
class Solution {
public:
vector<vector<char>> board;
int dx[4]={-1,0,1,0},dy[4]={0,1,0,-1};
int n,m;
void solve(vector<vector<char>>& _board) {
board=_board;
n=board.size(),m=board[0].size();
for(int i=0;i<n;i++)
{
if(board[i][m-1]=='O') dfs(i,m-1);
if(board[i][0]=='O') dfs(i,0);
}
for(int i=0;i<m;i++)
{
if(board[0][i]=='O') dfs(0,i);
if(board[n-1][i]=='O') dfs(n-1,i);
}
for(int i=0;i<n;i++)
{
for(int j=0;j<m;j++)
{
if(board[i][j]=='#') _board[i][j]='O';
else _board[i][j]='X';
}
}
}
void dfs(int x,int y)
{
board[x][y]='#';
for(int i=0;i<4;i++)
{
int a=x+dx[i],b=y+dy[i];
if(a>=0 && a<n && b>=0 && b<m && board[a][b]=='O')
dfs(a,b);
}
}
};