给定一个二维的矩阵,包含 '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',都替换成'Z',然后将二维矩阵中的所有'O'替换成'X',将所有'Z'还原替换成'O'
class Solution {
public:
void solve(vector<vector<char>>& board) {
if (board.empty())
return;
if (board[0].empty())
return;
for(int i = 0; i < board.size(); ++i)
if(board[i][0] == 'O')
bfs(board, i, 0);
for(int i = 0; i < board.size(); ++i)
if(board[i][board[0].size() - 1] == 'O')
bfs(board, i, board[0].size() - 1);
for(int i = 0; i < board[0].size(); ++i)
if(board[0][i] == 'O')
bfs(board, 0, i);
for(int i = 0; i < board[0].size(); ++i)
if(board[board.size() - 1][i] == 'O')
bfs(board, board.size() - 1, i);
for(int i = 0; i < board.size(); ++i){
for(int j = 0; j < board[0].size(); ++j){
if('Z' == board[i][j])
board[i][j] = 'O';
else if('O' == board[i][j])
board[i][j] = 'X';
}
}
}
void bfs(vector<vector<char>>& board, int i, int j){
board[i][j] = 'Z';
if (i > 0 && board[i - 1][j] == 'O')
bfs(board, i - 1, j);
if (i < board.size() - 1 && board[i + 1][j] == 'O')
bfs(board, i + 1, j);
if (j > 0 && board[i][j - 1] == 'O')
bfs(board, i, j - 1);
if (j < board[0].size() - 1 && board[i][j + 1] == 'O')
bfs(board, i, j + 1);
}
};