1. AI技术聚合首页
  2. IT

算法沉淀——BFS 解决 FloodFill 算法（leetcode真题剖析）

社会演员多 • • IT • 阅读 26

算法沉淀——BFS 解决 FloodFill 算法

  • 01.图像渲染
  • 02.岛屿数量
  • 03.岛屿的最大面积
  • 04.被围绕的区域

BFS（广度优先搜索）解决 Flood Fill 算法的基本思想是通过从起始点开始，逐层向外扩展，访问所有与起始点相连且具有相同特性（颜色等）的区域。在 Flood Fill 中，通常是通过修改图像的像素颜色。

下面是 BFS 解决 Flood Fill 算法的步骤：

  1. 初始化： 将起始点的颜色修改为新的颜色，将起始点加入队列。
  2. BFS 遍历： 使用队列进行 BFS 遍历。每次从队列中取出一个位置，检查其相邻的位置是否符合条件（与起始点颜色相同），如果符合，则修改颜色并将其加入队列。这样，不断扩展遍历。
  3. 遍历直到完成： 重复上述步骤，直到队列为空，即没有可继续扩展的位置为止。此时，所有与起始点相连的区域都被成功修改。

在 Flood Fill 中，BFS 保证了相邻区域的逐层遍历，确保了所有相连的、颜色相同的区域都被填充为新的颜色。

01.图像渲染

题目链接：https://leetcode.cn/problems/flood-fill/

有一幅以 m x n 的二维整数数组表示的图画 image ，其中 image[i][j] 表示该图画的像素值大小。

你也被给予三个整数 sr , sc 和 newColor 。你应该从像素 image[sr][sc] 开始对图像进行 上色填充 。

为了完成 上色工作 ，从初始像素开始，记录初始坐标的 上下左右四个方向上 像素值与初始坐标相同的相连像素点，接着再记录这四个方向上符合条件的像素点与他们对应 四个方向上 像素值与初始坐标相同的相连像素点，……，重复该过程。将所有有记录的像素点的颜色值改为 newColor 。

最后返回 经过上色渲染后的图像 。

示例 1:

输入: image = [[1,1,1],[1,1,0],[1,0,1]]，sr = 1, sc = 1, newColor = 2
输出: [[2,2,2],[2,2,0],[2,0,1]]
解析: 在图像的正中间，(坐标(sr,sc)=(1,1)),在路径上所有符合条件的像素点的颜色都被更改成2。
注意，右下角的像素没有更改为2，因为它不是在上下左右四个方向上与初始点相连的像素点。

示例 2:

输入: image = [[0,0,0],[0,0,0]], sr = 0, sc = 0, newColor = 2
输出: [[2,2,2],[2,2,2]]

思路

这个题我们可以使用最朴素的bfs遍历来解决。

代码

class Solution {
    const int dx[4] = {0, 0, 1, -1};  // 表示上、下、左、右四个方向的相对坐标变化
    const int dy[4] = {-1, 1, 0, 0};
public:
    vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int color) {
        int prev = image[sr][sc];  // 记录起始位置的颜色
        if (prev == color) return image;  // 如果新旧颜色相同，不需要进行填充
        int m = image.size(), n = image[0].size();

        queue<pair<int, int>> q;
        q.push({sr, sc});

        while (!q.empty()) {
            auto [a, b] = q.front();
            q.pop();

            image[a][b] = color;  // 修改当前位置的颜色

            for (int i = 0; i < 4; ++i) {
                int x = a + dx[i], y = b + dy[i];
                // 判断新的坐标是否越界，并且颜色与旧颜色相同
                if (x >= 0 && x < m && y >= 0 && y < n && image[x][y] == prev) {
                    q.push({x, y});
                }
            }
        }
        return image;
    }
};
  1. 初始化： 记录起始位置的颜色 prev，如果起始颜色和目标颜色相同，直接返回原图。
  2. BFS 遍历： 使用队列 q 进行 BFS 遍历。从起始位置开始，逐层遍历相邻位置，将颜色修改为目标颜色。
  3. 遍历直到完成： 重复上述步骤，直到队列为空，即没有可继续扩展的位置为止。此时，所有与起始点相连的区域都被成功修改为新的颜色。

02.岛屿数量

题目链接：https://leetcode.cn/problems/number-of-islands/

给你一个由 '1'（陆地）和 '0'（水）组成的的二维网格，请你计算网格中岛屿的数量。

岛屿总是被水包围，并且每座岛屿只能由水平方向和/或竖直方向上相邻的陆地连接形成。

此外，你可以假设该网格的四条边均被水包围。

示例 1：

输入：grid = [
  ["1","1","1","1","0"],
  ["1","1","0","1","0"],
  ["1","1","0","0","0"],
  ["0","0","0","0","0"]
]
输出：1

示例 2：

输入：grid = [
  ["1","1","0","0","0"],
  ["1","1","0","0","0"],
  ["0","0","1","0","0"],
  ["0","0","0","1","1"]
]
输出：3

提示：

  • m == grid.length
  • n == grid[i].length
  • 1 <= m, n <= 300
  • grid[i][j] 的值为 '0' 或 '1'

思路

使用bfs思想遍历每一个方格，将一块遍历过的岛屿全都标记为海洋也就是0，或者使用同等大小的数组进行遍历的标记。

代码

class Solution {
    const int dx[4] = {0, 0, 1, -1};  // 表示上、下、左、右四个方向的相对坐标变化
    const int dy[4] = {1, -1, 0, 0};
    int m, n;  // m表示行数，n表示列数
    queue<pair<int, int>> q;  // 用于BFS的队列

    // BFS函数，从起点 (i, j) 开始遍历并标记属于同一岛屿的所有位置
    void bfs(vector<vector<char>>& grid, int i, int j) {
        q.push({i, j});  // 将起点入队
        grid[i][j] = '0';  // 标记已经遍历过的位置
        while (!q.empty()) {
            auto [a, b] = q.front();
            q.pop();
            for (int k = 0; k < 4; ++k) {
                int x = a + dx[k], y = b + dy[k];
                // 判断新的坐标是否越界，并且是岛屿的一部分
                if (x >= 0 && x < m && y >= 0 && y < n && grid[x][y] == '1') {
                    q.push({x, y});  // 将相邻的岛屿位置入队
                    grid[x][y] = '0';  // 标记已经遍历过的位置
                }
            }
        }
    }

public:
    int numIslands(vector<vector<char>>& grid) {
        m = grid.size();  // 获取行数
        n = grid[0].size();  // 获取列数
        int ret = 0;  // 记录岛屿数量

        // 遍历整个网格
        for (int i = 0; i < m; ++i) {
            for (int j = 0; j < n; ++j) {
                if (grid[i][j] == '1') {
                    ret++;  // 发现新的岛屿，增加计数
                    bfs(grid, i, j);  // 使用BFS遍历并标记所有属于同一岛屿的位置
                }
            }
        }

        return ret;  // 返回岛屿数量
    }
};
  1. 初始化： 定义了方向数组 dx 和 dy，表示上、下、左、右四个方向的相对坐标变化。初始化队列 q，用于BFS遍历。
  2. BFS遍历： 对于每个未被访问的岛屿起点，调用 bfs 函数进行BFS遍历。在BFS过程中，将属于同一岛屿的位置标记为已访问。
  3. 遍历整个网格： 使用两层循环遍历整个网格，如果发现未访问过的岛屿起点，就调用 bfs 函数进行遍历，并增加岛屿数量计数。
  4. 返回结果： 最终返回岛屿的数量。

03.岛屿的最大面积

给你一个大小为 m x n 的二进制矩阵 grid 。

岛屿 是由一些相邻的 1 (代表土地) 构成的组合，这里的「相邻」要求两个 1 必须在 水平或者竖直的四个方向上 相邻。你可以假设 grid 的四个边缘都被 0（代表水）包围着。

岛屿的面积是岛上值为 1 的单元格的数目。

计算并返回 grid 中最大的岛屿面积。如果没有岛屿，则返回面积为 0 。

示例 1：

输入：grid = [[0,0,1,0,0,0,0,1,0,0,0,0,0],[0,0,0,0,0,0,0,1,1,1,0,0,0],[0,1,1,0,1,0,0,0,0,0,0,0,0],[0,1,0,0,1,1,0,0,1,0,1,0,0],[0,1,0,0,1,1,0,0,1,1,1,0,0],[0,0,0,0,0,0,0,0,0,0,1,0,0],[0,0,0,0,0,0,0,1,1,1,0,0,0],[0,0,0,0,0,0,0,1,1,0,0,0,0]]
输出：6
解释：答案不应该是 11 ，因为岛屿只能包含水平或垂直这四个方向上的 1 。

示例 2：

输入：grid = [[0,0,0,0,0,0,0,0]]
输出：0

提示：

  • m == grid.length
  • n == grid[i].length
  • 1 <= m, n <= 50
  • grid[i][j] 为 0 或 1

思路

总体和上面的题目解法一样，只不过我们在遍历每个岛屿时，顺便计算个数即岛屿的面积，每次计算完再比较，最后返回最大的岛屿面积。

代码

class Solution {
    const int dx[4] = {0, 0, 1, -1};  // 上、下、右、左四个方向的相对坐标变化
    const int dy[4] = {-1, 1, 0, 0};
    queue<pair<int, int>> q;  // 用于BFS的队列
    int m, n;  // m 表示行数，n 表示列数

    // BFS 函数，从起点 (i, j) 开始遍历并标记属于同一岛屿的所有位置
    int bfs(vector<vector<int>>& grid, int i, int j) {
        int count = 1;  // 记录岛屿的大小
        q.push({i, j});  // 将起点入队
        grid[i][j] = 0;  // 标记已经遍历过的位置

        while (!q.empty()) {
            auto [a, b] = q.front();
            q.pop();

            for (int k = 0; k < 4; ++k) {
                int x = a + dx[k], y = b + dy[k];
                // 判断新的坐标是否越界，并且是岛屿的一部分
                if (x >= 0 && x < m && y >= 0 && y < n && grid[x][y] == 1) {
                    grid[x][y] = 0;  // 标记已经遍历过的位置
                    count++;  // 增加岛屿的大小
                    q.push({x, y});  // 将相邻的岛屿位置入队
                }
            }
        }
        return count;
    }

public:
    int maxAreaOfIsland(vector<vector<int>>& grid) {
        int ret = 0;  // 记录最大岛屿面积
        m = grid.size();  // 获取行数
        n = grid[0].size();  // 获取列数

        // 遍历整个网格
        for (int i = 0; i < m; ++i) {
            for (int j = 0; j < n; ++j) {
                if (grid[i][j] == 1) {
                    ret = max(ret, bfs(grid, i, j));  // 计算并更新最大岛屿面积
                }
            }
        }

        return ret;  // 返回最大岛屿面积
    }
};
  1. 初始化： 定义了方向数组 dx 和 dy，表示上、下、左、右四个方向的相对坐标变化。初始化队列 q，用于BFS遍历。
  2. BFS遍历： 对于每个未被访问的岛屿起点，调用 bfs 函数进行BFS遍历。在BFS过程中，将属于同一岛屿的位置标记为已访问，并统计岛屿的大小。
  3. 遍历整个网格： 使用两层循环遍历整个网格，如果发现未访问过的岛屿起点，就调用 bfs 函数进行遍历，并计算并更新最大岛屿面积。
  4. 返回结果： 最终返回最大岛屿面积。

04.被围绕的区域

题目链接：https://leetcode.cn/problems/surrounded-regions/

给你一个 m x n 的矩阵 board ，由若干字符 'X' 和 'O' ，找到所有被 'X' 围绕的区域，并将这些区域里所有的 'O' 用 'X' 填充。

示例 1：

输入：board = [["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'。如果两个元素在水平或垂直方向相邻，则称它们是“相连”的。

示例 2：

输入：board = [["X"]]
输出：[["X"]]

提示：

  • m == board.length
  • n == board[i].length
  • 1 <= m, n <= 200
  • board[i][j] 为 'X' 或 'O'

思路

这里我们可以使用bfs先处理所有与边界有关的位置，把它修改成其他字符，再依次遍历，留在格子中的字母O就是需要被修改的，而改成其他字符的就可以改回字母O。

代码

class Solution {
    const int dx[4] = {0, 0, 1, -1};  // 上、下、右、左四个方向的相对坐标变化
    const int dy[4] = {-1, 1, 0, 0};
    int m, n;  // m 表示行数，n 表示列数
    queue<pair<int, int>> q;  // 用于BFS的队列

    // BFS 函数，从起点 (i, j) 开始遍历并标记属于同一区域的所有位置
    void bfs(vector<vector<char>>& board, int i, int j) {
        q.push({i, j});  // 将起点入队
        board[i][j] = '#';  // 标记已经遍历过的位置

        while (!q.empty()) {
            auto [a, b] = q.front();
            q.pop();

            for (int k = 0; k < 4; ++k) {
                int x = a + dx[k], y = b + dy[k];
                // 判断新的坐标是否越界，并且是未被围绕的区域
                if (x >= 0 && x < m && y >= 0 && y < n && board[x][y] == 'O') {
                    board[x][y] = '#';  // 标记已经遍历过的位置
                    q.push({x, y});  // 将相邻的未被围绕的区域位置入队
                }
            }
        }
    }

public:
    void solve(vector<vector<char>>& board) {
        m = board.size();  // 获取行数
        n = board[0].size();  // 获取列数

        // 对边界上的'O'进行BFS遍历，标记为'#'
        for (int i = 0; i < n; ++i) {
            if (board[0][i] == 'O') bfs(board, 0, i);
            if (board[m - 1][i] == 'O') bfs(board, m - 1, i);
        }
        for (int i = 1; i < m - 1; ++i) {
            if (board[i][0] == 'O') bfs(board, i, 0);
            if (board[i][n - 1] == 'O') bfs(board, i, n - 1);
        }

        // 遍历整个网格，将未被标记的'O'修改为'X'，已经标记的'#'修改回'O'
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (board[i][j] == '#') board[i][j] = 'O';
                else if (board[i][j] == 'O') board[i][j] = 'X';
            }
        }
    }
};

版权声明：本文为博主作者：爱学习的鱼佬原创文章，版权归属原作者，如果侵权，请联系我们删除！

原文链接：https://blog.csdn.net/kingxzq/article/details/136129927

共计人评分，平均分

到目前为止还没有投票！成为第一位评论此文章。

赞 (0)
社会演员多的头像社会演员多普通用户
0
上一篇 2024年2月19日
下一篇 2024年2月19日

相关推荐