算法题九 之 被围绕的区域

题目

给定一个二维的矩阵，包含 ‘X’ 和 ‘O’（字母 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

思路

• 第一步：从题目的观察中可以得知，只要O与边界连接，这个O便不被包围。而与O这个连接的所有O，也都不被包围。
• 第二步：根据第一步的观察，只要O通过连接链与边界产生关系，则当前O是不被包围，反之，则被包围。
• 第三步：正向思路是遍历整个矩阵，遇到O做深度遍历，看看是否与边界相连。反向思路则是遍历边界，遇到O做深度遍历，把所有与边界的O都找出来，做一个标记。没有被标记的O，则都是被包围的。

附上代码：

public void solve(char[][] board) {

    if(board == null || board.length == 0){
        return;
    }

    int n = board.length;
    int m = board[0].length;
    boolean[][] states = new boolean[n][m];
    // 与边界相连接的地方 则是可以保留的地方
    // 从左上到左下
    for(int i = 0; i < n; i++){
        if(board[i][0] == 'O' && !states[i][0]){
            dfs(i,0,board,states);
        }
    }

    // 从右上到右下
    for(int i = 0; i < n; i++){
        if(board[i][m - 1] == 'O' && !states[i][m - 1]){
            dfs(i,m - 1,board,states);
        }
    }

    // 从左上到右上
    for(int i = 0; i < m; i++){
        if(board[0][i] == 'O' && !states[0][i]){
            dfs(0,i,board,states);
        }
    }

    // 从左下到右下
    for(int i = 0; i < m; i++){
        if(board[n - 1][i] == 'O' && !states[n - 1][i]){
            dfs(n - 1,i,board,states);
        }
    }

    for(int i = 0; i < n; i++) {
        for(int j = 0; j < m; j++){
            if(board[i][j] == 'O' && !states[i][j]){
                board[i][j] = 'X';
            }
        }
    }

}

public void dfs(int i,int j,char[][] board,boolean[][] states){

    if(states[i][j]){
        return;
    }
    states[i][j] = true;
    // 向左
    if(j - 1 >= 0 && !states[i][j - 1] && board[i][j - 1] == 'O'){
        dfs(i,j - 1,board,states);
    }
    // 向右
    if(j + 1 < board[i].length && !states[i][j + 1] && board[i][j + 1] == 'O'){
        dfs(i,j + 1,board,states);
    }

    // 向上
    if(i - 1 >= 0 && !states[i - 1][j] && board[i - 1][j] == 'O'){
        dfs(i - 1,j,board,states);
    }

    // 向下
    if(i + 1 < board.length && !states[i + 1][j] && board[i + 1][j] == 'O'){
        dfs(i + 1,j,board,states);
    }
}