问题其实就是如何判断一个 ‘O’ 是否是被 ‘X’ 围绕的:
- 被围绕的区间显然不会存在于边界上,换句话说,任何边界上的 ‘O’ 都不会被填充为 ‘X’。
- 而任何不在边界上,或不与边界上的 ‘O’ 相连的 ‘O’ 最终都会被填充为 ‘X’。
那我们其实就只需要将与边界相连的 ‘O’ 找出来,然后把剩下的 ‘O’ 全部修改为 ‘X’ 就可以了。一共需要考虑四个边界,上下左右:
dfs遍历:
class Solution {
public:
int m, n;
void dfs(vector<vector<char>>& board, int x, int y){
if(x < 0 || x >= m || y < 0 || y >= n || board[x][y] != 'O') return;
board[x][y] = '-';
//其实就是检查上下左右是否有相连的字符'O'
//函数的设计很好地避开了边界检查
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) {
m = board.size();
if(m == 0) return;
n = board[0].size();
if(m < 3 || n < 3) return; //无法形成围绕
for(int i = 0; i < m; ++i){
dfs(board, i, 0);//左边界
dfs(board, i, n-1);//右边界
}
for(int i = 1; i < n-1; ++i){//从第一列开始到倒数第二列
dfs(board, 0, i);//上边界
dfs(board, m-1, i);//下边界
}
for(int i = 0; i < m; ++i){
for(int j = 0; j < n; ++j){
if(board[i][j] == '-') board[i][j] = 'O';
else if(board[i][j] == 'O') board[i][j] = 'X';
}
}
}
};