给定一个二维的矩阵,包含 '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或者BFS找边界上的‘O'相连的‘O',标记为T,然后再遍历整个矩阵,被标记过的就是边界上的,仍然是‘O‘,否则'O'要变为'X'
也可用并查集,并查集一般用来解决连通区域的问题,这里可以把边界上的'O'以及与它们相连的'O'看作一个连通区域,它们的father是同一个,被'X'包围的'O'属于另外的连通区域,生成矩阵的时候,属于边界'O'连通区域的‘O'不变,另外的'O'变为'X'
class Solution {
private:
vector<vector<int>> direction = {{-1,0},{1,0},{0,-1},{0,1}};
void dfs(vector<vector<char>>& board, int x, int y)
{
int row = board.size();
int col = board[0].size();
if(x < 0 || x >= row || y < 0 || y >= col || board[x][y] != 'O')
return ;
board[x][y] = 'T';
for(int i = 0; i < 4; i++)
{
int nx = x + direction[i][0];
int ny = y + direction[i][1];
dfs(board, nx, ny);
}
}
public:
void solve(vector<vector<char>>& board) {
int row = board.size();
if(!row) return;
int col = board[0].size();
//标记边界上的'0'以及与它们直接或间接相连的'0'
for(int i = 0; i < row; i++)
{
dfs(board, i, 0);
dfs(board, i, col - 1);
}
for(int j = 0; j < col; j++)
{
dfs(board, 0, j);
dfs(board, row - 1, j);
}
for(int i = 0; i < row; i++)
{
for(int j = 0; j < col; j++)
{
if(board[i][j] == 'T')
board[i][j] = 'O';
else if(board[i][j] == 'O')
board[i][j] = 'X';
}
}
}
};