单源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;
}