注意到题目解释中提到:任何边界上的 O 都不会被填充为 X。 我们可以想到,所有的不被包围的 O 都直接或间接与边界上的 O 相连。我们可以利用这个性质判断 O 是否在边界上,具体地说:
对于每一个边界上的 O,我们以它为起点,标记所有与它直接或间接相连的字母 O(标记为‘A’);
最后我们遍历这个矩阵,对于每一个字母:
如果该字母被标记过,则该字母为没有被字母 X 包围的字母 O,我们将其还原为字母 O;
如果该字母没有被标记过,则该字母为被字母 X 包围的字母 O,我们将其修改为字母 X。
class Solution {
private:
int foot[4][2]={1,0,-1,0,0,1,0,-1};
void dfs(vector<vector<char>>& v,int i,int j)
{
for (int ii=0;ii<4;ii++)
{
int x=i+foot[ii][0];
int y=j+foot[ii][1];
if (x>=0 && x<v.size() && y>=0 && y<v[0].size())
if (v[x][y]=='O')
{
v[x][y]='A';
dfs(v,x,y);
}
}
}
public:
void solve(vector<vector<char>>& board) {
if (board.size()==0 || board[0].size()==0) return;
int n=board.size(),m=board[0].size();
for (int i=0;i<n;i++)
for (int j=0;j<m;j++)
if (i==0 || i==n-1 || j==0 || j==m-1)
if (board[i][j]=='O')
{
board[i][j]='A';
dfs(board,i,j);
}
for (int i=0;i<n;i++)
for (int j=0;j<m;j++)
if (board[i][j]=='A')
board[i][j]='O';
else if (board[i][j]=='O')
board[i][j]='X';
}
};
这边记录官方题解的广搜写法,主要注意
queue<pair<int, int>> que;
que.emplace(0, i);
int x = que.front().first, y = que.front().second;
的使用
class Solution {
public:
int n, m;
void dfs(vector<vector<char>>& board, int x, int y) {
if (x < 0 || x >= n || y < 0 || y >= m || board[x][y] != 'O') {
return;
}
board[x][y] = 'A';
dfs(board, x + 1, y);
dfs(board, x - 1, y);
dfs(board, x, y + 1);
dfs(board, x, y - 1);
}
void solve(vector<vector<char>>& board) {
n = board.size();
if (n == 0) {
return;
}
m = board[0].size();
for (int i = 0; i < n; i++) {
dfs(board, i, 0);
dfs(board, i, m - 1);
}
for (int i = 1; i < m - 1; i++) {
dfs(board, 0, i);
dfs(board, n - 1, i);
}
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
if (board[i][j] == 'A') {
board[i][j] = 'O';
} else if (board[i][j] == 'O') {
board[i][j] = 'X';
}
}
}
}
};