leetcode 130.被围绕的区域 Java

题目链接

https://leetcode-cn.com/problems/surrounded-regions/

描述

给定一个二维的矩阵，包含 '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'。如果两个元素在水平或垂直方向相邻，则称它们是“相连”的。

初始代码模板

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

    }
}

代码

初看晦涩难懂，其实转换个思路，就很简单了。
和边界相连的’O’肯定是无法填充的，所以可以把与边界相连的’O’区间先用特殊字符标识一下，比如换成’#’，然后遍历一下数组，不是’#‘的换成’X’，而’#‘改回’O’，就是答案了，没有必要寻找被围绕的区间。

class Solution {
    
    private int[] xArr = {1, -1, 0, 0};
    private int[] yArr = {0, 0, 1, -1};

    public void solve(char[][] board) {
        if (board == null || board.length == 0 || board[0] == null || board[0].length == 0) {
            return;
        }

        int m = board.length;
        int n = board[0].length; 

        for (int i = 0; i < n; i++) {
            dfs(board, 0, i);
            dfs(board, m - 1, i);
        }

        for (int i = 0; i < m; i++) {
            dfs(board, i, 0);
            dfs(board, i, n - 1);
        }

        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                board[i][j] = board[i][j] == '#' ? 'O' : 'X';
            }
        }
    }

    private void dfs(char[][] board, int x, int y) {
        if (x < 0 || x >= board.length || y < 0 || y >= board[0].length || board[x][y] != 'O') {
            return;
        }

        board[x][y] = '#';

        for (int i = 0; i < 4; i++) {
            dfs(board, x + xArr[i], y + yArr[i]);
        }
    }
}