给你一个 m x n
的矩阵 board
,由若干字符 'X'
和 '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
m == board.length
n == board[i].length
1 <= m, n <= 200
board[i][j]
为'X'
或'O
刚开始想遍历整个 board 找到 o 进行 dfs搜索。但是发现如果 O 在边界时回溯不好写。
之后看官方解题代码,立马醒悟,可以反过来想,只要找到在边界的相连 O 剩下的就是不在边界的,所以代码也就好写了。
class Solution {
int n, m; //二维数组 长 宽
public void solve(char[][] board) {
n = board.length;
m = board[0].length;
//两个循环遍历边, dfs 每个边,将所有不需要变换的 O 转换为 A
for(int i=0; i<n; i++){
dfs(board,i,0);
dfs(board,i,m-1);
}
for(int j=1; j<m-1; j++){
dfs(board,0,j);
dfs(board,n-1,j);
}
//将所有 O 变为 x ,将所有 A 变为 O 即为最终答案
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
if(board[i][j] == 'O'){
board[i][j] = 'X';
}
if(board[i][j] == 'A'){
board[i][j] = 'O';
}
}
}
}
//将所有不需要变换的 O 变换为 A
public void dfs(char[][] board, int x, int y) {
//如果x,y超出数组边界,或者不为 O 则退出递归,如果该点为 A 也需要退出递归,避免死循环
if (x < 0 || x >= n || y < 0 || y >= m || board[x][y] != 'O' || board[x][y] == 'A') {
return;
}
//如果是 O 则变为 A ,并递归该点四周的点
board[x][y] = 'A';
dfs(board, x + 1, y);
dfs(board, x - 1, y);
dfs(board, x, y + 1);
dfs(board, x, y - 1);
}
}