多源BFS(图的广搜)

单源BFS就是树的广搜思路,把根节点入队,然后一层层的遍历
多源BFS就是图的广搜思路,可以多个源点入队,为防止某个点入队多次,需要标记,多源BFS一般用于解决下面这种多源点问题,会提高代码执行效率。

例题:矩阵距离

给定一个N行M列的01矩阵A,A[i][j] 与 A[k][l] 之间的曼哈顿距离定义为:
dist(A[i][j],A[k][l])=|i−k|+|j−l|
输出一个N行M列的整数矩阵B,其中:
B[i][j]=min1≤x≤N,1≤y≤M,A[x][y]=1dist(A[i][j],A[x][y])

输入格式

第一行两个整数n,m。

接下来一个N行M列的01矩阵,数字之间没有空格。

输出格式

一个N行M列的矩阵B,相邻两个整数之间用一个空格隔开。

数据范围

1≤N,M≤1000

输入样例:

3 4
0001
0011
0110

输出样例:

3 2 1 0
2 1 0 0
1 0 0 1

题意就是输出矩阵A中每个格子到最近的一个1的曼哈顿距离。

如何做?
如果我们从第一个点开始一个一个的搜,复杂度会是O(n^2),会TLE,而且题目明显是一个最短路问题,所有边权重都为1,这点不可改变,但是题目要求我们找到与每个点到点"1"的距离,就可以看作这是一个多起点问题,那我们把所有为“1”的点加入队列,根据BFS的单调性和两段性,从“1"开始求出每个“1”到每个点的距离,而且使用Flood Fill 解法(—>Flood Fill方法)会直接求出一个起点到其它位置的最小步数,这样就可以在O(n)的复杂度内求出矩阵距离。

#include <iostream>
#include <cstring>
#include <algorithm>

#define x first
#define y second
using namespace std;
typedef pair<int,int> PII;

const int N = 1010, M = N * N;
int n, m;
int dist[N][N];
PII q[M];
char g[N][N];
int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};

void bfs()
{
//将所有为“1”的点加入队列后进行广搜,然后Flood-fill求最小距离
    memset(dist, -1, sizeof dist);
    int hh = 0, tt = -1;
    
    for(int i = 0; i < n; i ++ )
        for(int j = 0; j < m; j ++ ) 
            if(g[i][j] == '1')
            {
                dist[i][j] = 0;
                q[++tt] = {i, j};
            }
            
    while(hh <= tt)
    {
        PII t = q[ hh ++ ];
        for(int i = 0; i < 4; i ++ )
        {
            int a = t.x + dx[i], b = t.y + dy[i];
            
            if( a < 0 || a >= n || b < 0 || b >= m) continue;
            if(dist[a][b] != -1) continue;
            
            dist[a][b] = dist[t.x][t.y] + 1;
            q[ ++ tt] = {a,b};
        }
    }
}
int main()
{
    scanf("%d%d", &n, &m);
    
    for(int i = 0; i< n; i ++ ) scanf("%s", &g[i]);
    
    bfs();
    
    for(int i = 0; i < n; i ++ )
    {
        for(int j = 0; j < m; j ++ )
            printf("%d ",dist[i][j]);
        puts("");
    }
    
    return 0;
}

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值