leetcode130. 被围绕的区域

偏执的太偏执、 2023-07-06 08:57 14阅读 0赞

给定一个二维的矩阵，包含 '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’就是和边界连接的。比如如下情况：

X X O X  
X O O X  
X X O X  
X O O X

'O'不会被改变，因为和边界的‘O’连接，没有被‘X’围绕。

所以，我们从边界开始dfs，把遇到的“O”变为“\#”代表和边界相连。

然后遍历数组，把''O'变为X（因为不和边界相连），把\#变为X（因为和边界相连）。

class Solution {
        public void solve(char[][] board) {
            if (board == null || board.length == 0) return;
            int m = board.length;
            int n = board[0].length;
            for (int i = 0; i < m; i++) {
                for (int j = 0; j < n; j++) {
                    // 从边缘开始搜索
                    if ((i == 0 || j == 0 || i == m - 1 || j == n - 1) && board[i][j] == 'O') {
                        dfs(board, i, j);
                    }
                }
            }
    
            for (int i = 0; i < m; i++) {
                for (int j = 0; j < n; j++) {
                    //没有被搜索到过，不和边界连接
                    if (board[i][j] == 'O') {
                        board[i][j] = 'X';
                    }
                    //被改变过，和边界连接
                    if (board[i][j] == '#') {
                        board[i][j] = 'O';
                    }
                }
            }
        }
    
        public void dfs(char[][] board, int i, int j) {
            if (i < 0 || j < 0 || i >= board.length  || j >= board[0].length || board[i][j] == 'X' || board[i][j] == '#') {
                // board[i][j] == '#' 说明已经搜索过了. 
                return;
            }
            board[i][j] = '#';
            dfs(board, i - 1, j); // 上
            dfs(board, i + 1, j); // 下
            dfs(board, i, j - 1); // 左
            dfs(board, i, j + 1); // 右
        }
    }