BFS在矩阵中的简单应用(单源 多源)

配套视频:https://www.bilibili.com/video/BV1Su41197rq?spm_id_from=333.999.0.0

模板

BFS有强烈的“步数”意识。每一步的搜索向外扩展一层。

BFS的核心是搜索:让每个点向所有可能的方向搜索。

核心模板如下(带注释)

while(q.size())
{
    int len = q.size();
    //依据题意总步数或其他
    while(len --)
    {
        auto [x, y] = q.front(); q.pop();
        //尝试向外拓展
        for(int i = 0; i < 方向总数; ++ i)
        {
            int xx = x + dx[i], yy = y + dy[i];
            //即将搜索的坐标越界或已经被搜索过
            if(xx < 0 || xx >= n || yy < 0 || yy >= m || b[xx][yy]) continue; 
            q.push({xx, yy}); b[xx][yy] = 1;
        }
    }
}

一、单源

例一、岛屿的最大面积

链接:https://leetcode-cn.com/problems/max-area-of-island/

题目

给你一个大小为 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

题目分析

经典的单源bfs。每个点的搜索方向只有$4$个

遍历一遍二维矩阵,某个点为$1$且未被访问过就从该点出发搜索。

参考代码(带注释)

class Solution {
public:
    int dx[4] = {0,0,1,-1}; int dy[4] = {1,-1,0,0}; //搜索的四个方向
    int maxAreaOfIsland(vector<vector<int>>& grid) {
        int n = grid.size(), m = grid[0].size();
        int maxn = 0;
        vector<vector<bool>> b(n, vector<bool>(m)); //布尔数组标记有没有被访问
        for(int i = 0; i < n; ++ i)
        {
            for(int j = 0; j < m; ++ j)
            {
                if(b[i][j]) continue;
                b[i][j] = true;
                if(!grid[i][j]) continue; //该点即未被访问又为1 那就从这个点开始搜索
                queue<pair<int, int>> q;
                q.push({i, j});
                int tmp = 0;
                while(q.size())
                {
                    int len = q.size();
                    tmp += len; //记录一个"1连接块"中"1"的个数
                    for(int k = 0; k < len; ++ k)
                    {
                        auto [x, y] = q.front();
                        q.pop();
                        for(int t = 0; t < 4; ++ t)
                        {
                            int xx = x + dx[t], yy = y + dy[t];
                            if(xx < 0 || xx >= n || yy < 0 || yy >= m || b[xx][yy] || !grid[xx][yy]) continue;
                            b[xx][yy] = true;
                            q.push({xx, yy}); //只有"1"才能进队列
                        }
                    }
                }
                maxn = max(maxn, tmp);
            }
        }

        return maxn;
    }
};

例二、地牢大师

链接:https://www.acwing.com/problem/content/1098/

题目

你现在被困在一个三维地牢中,需要找到最快脱离的出路!

地牢由若干个单位立方体组成,其中部分不含岩石障碍可以直接通过,部分包含岩石障碍无法通过。

向北,向南,向东,向西,向上或向下移动一个单元距离均需要一分钟。

你不能沿对角线移动,迷宫边界都是坚硬的岩石,你不能走出边界范围。

请问,你有可能逃脱吗?

如果可以,需要多长时间?

输入格式

输入包含多组测试数据。

每组数据第一行包含三个整数 L,R,C

分别表示地牢层数,以及每一层地牢的行数和列数。

接下来是 LRC 列的字符矩阵,用来表示每一层地牢的具体状况。

每个字符用来描述一个地牢单元的具体状况。

其中, 充满岩石障碍的单元格用”#”表示,不含障碍的空单元格用”.”表示,你的起始位置用”S”表示,终点用”E”表示。

每一个字符矩阵后面都会包含一个空行。

当输入一行为”0 0 0”时,表示输入终止。

输出格式

每组数据输出一个结果,每个结果占一行。

如果能够逃脱地牢,则输出”Escaped in x minute(s).”,其中X为逃脱所需最短时间。

如果不能逃脱地牢,则输出”Trapped!”。

数据范围

1≤L,R,C≤100

输入样例

3 4 5
S....
.###.
.##..
###.#

#####
#####
##.##
##...

#####
#####
#.###
####E

1 3 3
S##
#E#
###

0 0 0

输出样例

Escaped in 11 minute(s).
Trapped!

题目分析

起点和终点已定;方向确定;带有强烈的步数暗示。

和常规的bfs搜索方向不同,此题的搜索方向由二维变成三维。其它一样。

参考代码(带注释)

#include <bits/stdc++.h>
#define x first
#define y second

using namespace std;
int h, n, m;
const int N = 105;
char a[N][N][N];
bool b[N][N][N];
//搜索方向 六个
int dx[6] = {0,0,1,-1,0,0}; int dy[6] = {-1,1,0,0,0,0}; int dz[6] = {0,0,0,0,1,-1}; 
int st[3], en[3];

//要注意存储下标的意义
void solve()
{
    memset(b, 0, sizeof b);
    for(int i = 0; i < h; ++ i) for(int j = 0; j < n; ++ j) for(int k = 0; k < m; ++ k) //z x y
    {
        cin >> a[i][j][k];
        if(a[i][j][k] == 'S') st[2] = i, st[0] = j, st[1] = k, b[j][k][i] = true;
        else if(a[i][j][k] == 'E') en[2] = i, en[0] = j, en[1] = k;
        else if(a[i][j][k] == '#') b[j][k][i] = true;
    }
    queue<pair<pair<int, int>, int>>q;
    q.push({{st[0], st[1]}, st[2]}); //x y z
    int step = 0; bool flag = false;
    //常规bfs 只是换了下搜索方向
    while(q.size())
    {
        int len = q.size();
        step ++; //题目特性:记录时间 所以必须要每次拟清空队列
        while(len --)
        {
            auto it = q.front(); q.pop();
            int x = it.x.x, y = it.x.y, z = it.y;
            for(int i = 0; i < 6; ++ i)
            {
                int xx = x + dx[i], yy = y + dy[i], zz = z + dz[i];
                if(xx < 0 || xx >= n || yy < 0 || yy >= m || zz < 0 || zz >= h || b[xx][yy][zz]) continue;
                if(xx == en[0] && yy == en[1] && zz == en[2]) 
                {
                    flag = true;
                    break;
                }
                b[xx][yy][zz] = 1;
                q.push({{xx, yy}, zz});
            }
            if(flag) break;
        }
        if(flag) break;
    }
    if(flag) printf("Escaped in %d minute(s).\n", step);
    else puts("Trapped!");
}

int main()
{
    while(cin >> h >> n >> m && h) solve();
    return 0;
}

二、多源

与单源的区别

什么情况下用多源同时遍历?某些点有共同的特性以至于要从这些点同时开始

例题 腐烂的橘子

链接:https://leetcode-cn.com/problems/rotting-oranges

题目

在给定的 m x n 网格 grid 中,每个单元格可以有以下三个值之一:

值 0 代表空单元格;
值 1 代表新鲜橘子;
值 2 代表腐烂的橘子。

每分钟,腐烂的橘子 周围 4 个方向上相邻 的新鲜橘子都会腐烂。

返回 直到单元格中没有新鲜橘子为止所必须经过的最小分钟数。如果不可能,返回 -1 。

示例 1:

输入:grid = [[2,1,1],[1,1,0],[0,1,1]]
输出:4

示例 2:

输入:grid = [[2,1,1],[0,1,1],[1,0,1]]
输出:-1
解释:左下角的橘子(第 2 行, 第 0 列)永远不会腐烂,因为腐烂只会发生在 4 个正向上。

示例 3:

输入:grid = [[0,2]]
输出:0
解释:因为 0 分钟时已经没有新鲜橘子了,所以答案就是 0 。

提示:

m == grid.length
n == grid[i].length
1 <= m, n <= 10
grid[i][j] 仅为 0、1 或 2

题目分析

和“时间”挂钩;多个腐烂的橘子同时向四周腐烂。

常规的bfs基础上多点同时开始即可

参考代码(带注释)

class Solution {
public:
    int dx[4] = {0, 0, 1, -1}; int dy[4] = {1, -1, 0, 0};
    int orangesRotting(vector<vector<int>>& grid) {
        int n = grid.size(), m = grid[0].size();
        vector<vector<bool>>b(n, vector<bool>(m));
        int count = 0, sum = 0, zi = 0;
        queue<pair<int, int>>q;
        for(int i = 0; i < n; ++i){
            for(int j = 0; j < m; ++j){
                if(grid[i][j] == 2){
                    b[i][j] = true;
                    q.push({i, j});
                    ++zi;
                }
                else if(grid[i][j] == 1)
                    ++sum;
            }
        }
        if(sum + zi == 0) return 0; //没有橘子

        int step = -1;
        while(!q.empty()){
            step ++;
            int len = q.size();
            while(len --)
            {
                auto [x, y] = q.front(); q.pop();
                for(int i = 0; i < 4; ++ i)
                {
                    int xx = x + dx[i], yy = y + dy[i];
                    if(xx < 0 || xx >= n || yy < 0 || yy >= m || b[xx][yy] || !grid[xx][yy]) continue;
                    count ++;
                    b[xx][yy] = true;
                    q.push({xx, yy});
                }
            }
        }
        return (count == sum) ? step : -1;
    }
};

总结

模板会了后和二维矩阵相关的裸题很简单,但有时候就是看不出来题目在考哪个知识点。

以上是蒟蒻写的简单bfs。希望能帮助到萌新,欢迎菊苣们指正。

暂无评论

发送评论 编辑评论


				
|´・ω・)ノ
ヾ(≧∇≦*)ゝ
(☆ω☆)
(╯‵□′)╯︵┴─┴
 ̄﹃ ̄
(/ω\)
∠( ᐛ 」∠)_
(๑•̀ㅁ•́ฅ)
→_→
୧(๑•̀⌄•́๑)૭
٩(ˊᗜˋ*)و
(ノ°ο°)ノ
(´இ皿இ`)
⌇●﹏●⌇
(ฅ´ω`ฅ)
(╯°A°)╯︵○○○
φ( ̄∇ ̄o)
ヾ(´・ ・`。)ノ"
( ง ᵒ̌皿ᵒ̌)ง⁼³₌₃
(ó﹏ò。)
Σ(っ °Д °;)っ
( ,,´・ω・)ノ"(´っω・`。)
╮(╯▽╰)╭
o(*////▽////*)q
>﹏<
( ๑´•ω•) "(ㆆᴗㆆ)
😂
😀
😅
😊
🙂
🙃
😌
😍
😘
😜
😝
😏
😒
🙄
😳
😡
😔
😫
😱
😭
💩
👻
🙌
🖕
👍
👫
👬
👭
🌚
🌝
🙈
💊
😶
🙏
🍦
🍉
😣
Source: github.com/k4yt3x/flowerhd
颜文字
Emoji
小恐龙
花!
上一篇
下一篇