DFS解决LeetCode 130被围绕的区域

题目描述

给你一个 m x n 的矩阵 board ,由若干字符 ‘X’ 和 ‘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’。如果两个元素在水平或垂直方向相邻,则称它们是“相连”的。

解题思路

DFS深度优先搜索,先找四条边上的O,因为4边上的O肯定不是被X包围的,剩下的就进行遍历,将O改为X即可

代码实现

class Solution {
    // 方向数组
    int[][] nextP = {
   
   { 0, 1 }, { 1, 0 }, { 0, -1 }, { -1, 0 } };
    // DFS搜索与外界联通的 O
    public void dfs(char[][] board,int row,int col,int CurX,int CurY ) {
        // 标记当前位置,当前位置设置为 M
        board[CurX][CurY] = 'M';
        //book[CurX][CurY] = 1;
        // 搜索4个方向
        for (int i = 0; i < 4; i++) {
            int newX = CurX + nextP[i][0];
            int newY = CurY + nextP[i][1];
            // 判断是否越界
            if(newX >= row || newX < 0 || newY >= col || newY < 0) {
                continue;
            }
            // 如果没有越界,如果等于O就继续去寻找
            if(board[newX][newY] == 'O') {
                dfs(board,row,col,newX,newY);
            }
        }
    }

    public void solve(char[][] board) {
        if(board.length == 0) {
            return;
        }
        int row = board.length;
        int col = board[0].length;
        // 先搜寻四边的O
        // 第一行 和 最后 一行 row 列 col
        for (int j = 0;j < col;j++) {
            if(board[0][j] == 'O') {
                dfs(board,row,col,0,j);
            }
            if(board[row-1][j] == 'O') {
                dfs(board,row,col,row-1,j);
            }
        }
        // 第一列 和 最后 一列
        for (int i = 0; i < row ; i++) {
            if(board[i][0] == 'O') {
                dfs(board,row,col,i,0);
            }
            if(board[i][col-1] == 'O') {
                dfs(board,row,col,i,col-1);
            }
        }

        // 遍历剩下的 进行修改
        for (int i = 0; i < row; i++) {
            for (int j = 0; j < col; j++) {
                 if(board[i][j] == 'O') {
                    board[i][j] = 'X';
                }else if(board[i][j] == 'M') {
                     board[i][j] = 'O';
                 }
            }
        }
    }
}

猜你喜欢

转载自blog.csdn.net/Biteht/article/details/125157614