给定一个二维的矩阵,包含 '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'
。如果两个元素在水平或垂直方向相邻,则称它们是“相连”的。
class Solution {
public:
int m,n;
int arr[4][2] = {{0,-1}, {-1,0}, {0,1}, {1,0}};
void dfs(vector<vector<char>> &board, vector<vector<bool>> &help, int i, int j)
{
help[i][j] = true;
board[i][j] = 'X';
for(int x=0;x<4;++x)
{
int new_i = i+arr[x][0], new_j = j+arr[x][1];
if(new_i>-1 && new_i<m && new_j>-1 && new_j<n && board[new_i][new_j]=='O')
dfs(board, help, new_i, new_j);
}
}
void solve(vector<vector<char>>& board) {
m = board.size();
if(!m)
return;
n = board[0].size();
vector<vector<bool>> help(m,vector<bool>(n,false));//若为false则改为X
for(int i=0;i<m;i++)
if(board[i][0]=='O')
dfs(board, help, i, 0);
for(int i=0;i<m; i++)
if(board[i][n-1] == 'O')
dfs(board,help,i,n-1);
for(int j=1;j<n-1;j++)
if(board[0][j]=='O')
dfs(board,help,0,j);
for(int j=1;j<n-1;j++)
if(board[m-1][j]=='O')
dfs(board,help,m-1,j);
for(int i=0;i<m;i++)
for(int j=0;j<n;j++)
if(help[i][j]) board[i][j] = 'O';
else board[i][j] = 'X';
}
};