给定一个二维的矩阵,包含 ‘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 {
private:
void dfs(vector<vector<char>> &board, int row, int col, vector<int> &dx, vector<int> &dy)
{
board[row][col] = 'Y';
int bottom = board.size()-1;
int right = board[0].size()-1;
for(int i = 0; i < 4; i++) //相邻四个方向遍历
{
int tx = row + dx[i]; //注意此处不能用row += dx[i]
int ty = col + dy[i];
if(tx < 0 || tx > bottom || ty < 0 || ty > right)
{
continue;
}
if(board[tx][ty] == 'O') //注意此处不能与上面的判断平级,因为已经将'O'变为'Y'了
dfs(board, tx, ty, dx, dy);
}
}
public:
void solve(vector<vector<char>>& board) {
if(board.size() == 0 || board[0].size() == 0) return;
vector<int> dx = {1, -1, 0, 0};
vector<int> dy = {0, 0, 1, -1};
int left = 0, right = (int)board[0].size()-1;
int top = 0, bottom = (int)board.size()-1;
int i = top, j = left;
for(i = top; j <= right; j++) //左上-->右上,注意此处是right,防止二维矩阵只有一个导致遍历不到
{
if(board[i][j] == 'O')
{
dfs(board, i, j, dx, dy);
}
}
for(j = right; i <= bottom-1; i++) //右上->右下
{
if(board[i][j] == 'O')
{
dfs(board, i, j, dx, dy);
}
}
for(i = bottom; j >= left+1; j--) //右下->左下
{
if(board[i][j] == 'O')
{
dfs(board, i, j, dx, dy);
}
}
for(j = left ; i >= top+1; i--) //左下->左上
{
if(board[i][j] == 'O')
{
dfs(board, i, j, dx, dy);
}
}
for(i = top; i <= bottom; i++) //将所有的'Y'替换为'O'
{
for(j = left; j <= right; j++)
{
if(board[i][j] == 'Y')
{
board[i][j] = 'O';
}
else
{
board[i][j] = 'X'; //不符合条件的'O'替换为'X'
}
}
}
}
};
首先对边界上每一个’O’做深度优先搜索,将与其相连的所有’O’改为’Y’。然后遍历矩阵,将矩阵中所有’O’改为’X’,将矩阵中所有’Y’变为’O’