思路:
依题意,被围的区间’O’不会位于边界上,那么我们设立一个标记数组protect,从边界开始BFS或者DFS,与边界上的’O’相连的内部’O’都protect起来。
之后,再去遍历矩阵内部,未被保护的’O’均修改成’X’即可.
注意输入为空时的处理。
class Solution {
int row, col;
int dx[4] = {0, 1, 0, -1}, dy[4] = {1, 0, -1, 0};
vector<vector<int>> protect;
public:
void BFS(int i, int j, vector<vector<char>>& board)
{
queue<pair<int, int>> q;
q.push(make_pair(i, j));
// 从边界开始,按照代码逻辑,边界没必要保护
while(!q.empty())
{
auto next = q.front();
q.pop();
for(int i = 0; i < 4; ++i)
{
int I = next.first + dx[i], J = next.second + dy[i];
// 满足边界条件,且是O,且还未被保护
if(I >= 1 && I < row - 1 && J >= 1 && J < col - 1 && board[I][J] == 'O' && !protect[I][J])
{
q.push(make_pair(I, J));
protect[I][J] = 1;
}
}
}
}
void solve(vector<vector<char>>& board) {
if(!board.size()) return;
row = board.size(), col = board[0].size();
protect.resize(row);
for(int i = 0; i < row; ++i)
{
protect[i].resize(col);
fill(protect[i].begin(), protect[i].end(), 0);
}
// 先处理边界BFS
for(int i: {0, row - 1}) // 上下边界
for(int j = 0; j < col; ++j)
{
if(board[i][j] == 'O') BFS(i , j, board);
}
for(int j:{0, col - 1})
for(int i = 0; i < row; ++i)
{
if(board[i][j] == 'O') BFS(i , j, board);
}
for(int i = 1; i < row - 1; ++i)
for(int j = 1; j < col - 1; ++j)
if(board[i][j] == 'O' && !protect[i][j]) // 到这里在内层且未被保护的'O'要被修改
board[i][j] = 'X';
}
};