LeetCode 高级 - 被围绕的区域

被围绕的区域

给定一个二维的矩阵，包含 ‘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’ 都不会被填充为 ‘X’。 任何不在边界上，或不与边界上的 ‘O’ 相连的 ‘O’ 最终都会被填充为 ‘X’。如果两个元素在水平或垂直方向相邻，则称它们是“相连”的。

分析

和以前做过的 岛屿的个数有点相像，只不过那道题是遍历所有节点，而这道题只需要遍历四周节点即可了。

思路就是，回溯法，以四周的节点为起点，向周围扩散，遇到O则将其改编为+，当四周节点都便利完成后，将所有O都替换为X，再将所有+替换回O

代码

    class Solution {
        //遍历四周，将所有与“O”相连的“O”都标记出来，然后对整个矩阵未被标记的元素赋值为“X”即可
        public void solve(char[][] board) {
            if(board==null||board.length <= 1||board[0].length<=1){
                return;
            }
            for(int i=0;i<board.length;i++){
                if(board[i][0] == 'O'){
                    dfs(board,i,0);
                }
                if(board[i][board[0].length-1] == 'O'){
                    dfs(board,i,board[0].length-1);
                }
            }
            for(int j=0;j<board[0].length;j++){
                if(board[0][j] == 'O'){
                    dfs(board,0,j);
                }
                if(board[board.length-1][j] == 'O'){
                    dfs(board,board.length-1,j);
                }
            }
            for(int i=0;i<board.length;i++){
                for(int j=0;j<board[0].length;j++){
                    if(board[i][j]=='+'){
                        board[i][j]='O';
                    }else if(board[i][j]=='O'){
                        board[i][j]='X';
                    }
                }
            }
        }
        private void dfs(char[][] board,int row,int col){
            if(row<0||row>=board.length||col<0||col>=board[0].length){
                return;
            }
            if(board[row][col]!='O')
                return;
            board[row][col] = '+';
            dfs(board,row+1,col);
            dfs(board,row,col+1);
            dfs(board,row-1,col);
            dfs(board,row,col-1);
        }
    }