题目所属分类
洪水灌溉算法 先找到处于边界不会被填充的O
原题链接
给你一个 m x n 的矩阵 board ,由若干字符 ‘X’ 和 ‘O’ ,找到所有被 ‘X’ 围绕的区域,并将这些区域里所有的 ‘O’ 用 ‘X’ 填充。
代码案例:输入:board = [[“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’。如果两个元素在水平或垂直方向相邻,则称它们是“相连”的。
题解
class Solution {
int [] dx = {-1,0,1,0};
int [] dy = {0,1,0,-1};
public void solve(char[][] board) {
int m = board.length;
if(m == 0) return ;
int n = board[0].length ;
// board[0][i]上边界、下边界board[m-1][i]
for(int i = 0 ; i < n ; i++){
if(board[0][i] == 'O') dfs(board,0,i);
if(board[m-1][i] == 'O') dfs(board,m-1,i);
}
// board[i][0]左边界、右边界board[i][n-1]
for(int i = 0 ; i < m;i++){
if(board[i][0] == 'O')dfs(board,i,0);
if(board[i][n-1] == 'O')dfs(board,i,n-1);
}
for(int i = 0 ; i< m ;i++){
for(int j = 0 ; j < n ;j ++){
if(board[i][j] == '*') board[i][j] = 'O';
else board[i][j] = 'X';
}
}
}
public void dfs(char[][] board , int x , int y){
board[x][y] = '*';
for(int i = 0 ; i < 4 ; i++){
int a = x + dx[i] ;
int b = y + dy[i] ;
if(a < 0 || a >= board.length || b <0 || b >= board[0].length) continue;
if(board[a][b] == 'O') dfs(board,a,b);
}
}
}