【Java】被围绕的区域（DFS）

题目描述

给定一个二维的矩阵，包含 '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

解题思路

DFS+递归寻找和边界联通的'O'，其他的'O‘都替换成'X'

package com.company;

import java.util.Arrays;

public class xxoo {

    public static void main(String[] args){

        xxoo test = new xxoo();
        char[][] xxxooo = {{'X','X','X','X'},{'X','O','O','X'},{'X','X','O','X'},{'X','O','X','X'}};
        test.solve(xxxooo);
        System.out.print(Arrays.deepToString(xxxooo));
    }

    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++) {
                // 从边缘o开始搜索
                // 判断是否和边界联通
                // boolean这里只要满足以下四个条件之一就说明此时在边界上
                boolean isEdge = i == 0 || j == 0 || i == m - 1 || j == n - 1;
                if (isEdge && board[i][j] == 'O') {
                    dfs(board, i, j);
                }
            }
        }
        // 把不是'#‘的'O'改成'X'
        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';
                }
            }
        }

    }
    // dfs找和边界联通的'O'并变成'#'，最后不能被'X'替换，并且下一次就不会被搜索
    public static 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);
    }
}